Home
last modified time | relevance | path

Searched defs:shape (Results 1 – 24 of 24) sorted by relevance

/packages/modules/NeuralNetworks/common/types/src/
DOperationsUtils.cpp49 uint32_t getNumberOfElements(const Shape& shape) { in getNumberOfElements()
57 uint32_t getNumberOfElements(const Shape& shape, size_t firstAxisInclusive, in getNumberOfElements()
69 uint32_t getNumberOfDimensions(const Shape& shape) { in getNumberOfDimensions()
73 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx) { in getSizeOfDimension()
78 uint32_t hasKnownRank(const Shape& shape) { in hasKnownRank()
/packages/modules/NeuralNetworks/common/include/
DCpuOperationUtils.h35 inline tflite::Dims<4> convertShapeToDims(const Shape& shape) { in convertShapeToDims()
56 inline tflite::RuntimeShape convertShapeToTflshape(const Shape& shape) { in convertShapeToTflshape()
162 bool initialize(const T* data, const Shape& shape) { in initialize()
186 bool initialize(T* data, const Shape& shape) { in initialize()
DOperationsExecutionUtils.h79 inline bool handleNegativeAxis(const Shape& shape, int32_t* axis) { in handleNegativeAxis()
268 inline bool transposeFirstTwoDimensions(const T* buffer, const Shape& shape, T* transposedBuffer) { in transposeFirstTwoDimensions()
289 inline bool transposeFirstTwoDimensions(const Shape& shape, Shape* transposedShape) { in transposeFirstTwoDimensions()
DCpuExecutor.h76 Shape shape() const { in shape() function
/packages/modules/NeuralNetworks/common/cpu_operations/
DElementwise.cpp37 const Shape& shape, T* output) { in compute()
46 inline bool compute(IntermediateType func(IntermediateType), const T* input, const Shape& shape, in compute()
DLogicalNot.cpp29 bool compute(const bool8* input, const Shape& shape, bool8* output) { in compute()
DNeg.cpp33 inline bool compute(const T* input, const Shape& shape, T* output) { in compute()
DLogSoftmax.cpp34 inline bool compute(const T* input, const Shape& shape, T beta, uint32_t axis, T* output) { in compute()
DBatchMatmul.cpp75 Shape swapRowColumnDims(const Shape& shape) { in swapRowColumnDims()
DDensify.cpp47 uint64_t getFlattenedIndex(const std::vector<int32_t>& indices, const std::vector<uint32_t>& shape, in getFlattenedIndex()
DQuantizedLSTMTest.cpp34 std::vector<uint32_t> shape; member
DGenerateProposals.cpp507 uint8_t castTo(float val, const Shape& shape) { in castTo()
512 int8_t castTo(float val, const Shape& shape) { in castTo()
/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/
Dsub_v1_2.mod.py37 shape = "{2, 4, 16, 2}, 0.5, 0" variable
/packages/modules/OnDevicePersonalization/src/com/android/ondevicepersonalization/services/inference/
DIsolatedModelServiceImpl.java125 int[] shape = tensor.shape(); in runTfliteInterpreter() local
/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/
Dsub_quant8_signed.mod.py86 shape = "{2, 4, 16, 2}, 0.5, -128" variable
/packages/modules/NeuralNetworks/runtime/test/
DTestExecution.cpp112 V1_2::OutputShape shape = {.dimensions = {1}, .isSufficient = false}; in execute_1_2() local
134 V1_2::OutputShape shape = {.dimensions = {1}, .isSufficient = false}; in execute_1_3() local
151 V1_2::OutputShape shape = {.dimensions = {1}, .isSufficient = false}; in executeSynchronously() local
171 V1_2::OutputShape shape = {.dimensions = {1}, .isSufficient = false}; in executeSynchronously_1_3() local
DTestValidateOperations.cpp1681 ANeuralNetworksOperandType shape = getOpType(ANEURALNETWORKS_TENSOR_INT32, 1, shapeDims); in reshapeOpTest() local
/packages/modules/NeuralNetworks/runtime/operation_converters/
DSubGraphContext.cpp201 std::vector<int32_t> shape(operand.dimensions.begin(), operand.dimensions.end()); in createTensorFlatbufferFromOperand() local
/packages/apps/Launcher3/src/com/android/launcher3/allapps/
DSectionDecorationHandler.java137 GradientDrawable shape = new GradientDrawable(); in applyBackground() local
/packages/modules/NeuralNetworks/runtime/
DExecutionBuilder.cpp71 [](const OutputShape& shape) { return shape.isSufficient; })) in validateOutputShapesFromDriver()
93 [](const OutputShape& shape) { return !shape.isSufficient; })) in validateOutputShapesFromDriver()
1252 for (const auto& shape : from) { in updateOutputShapes() local
/packages/apps/Dialer/java/com/android/dialer/lettertile/
DLetterTileDrawable.java379 @Shape final int shape, in setCanonicalDialerLetterTileDetails()
/packages/services/Car/car-lib/src/android/car/navigation/
Dnavigation_state.proto480 Shape shape = 1; field
/packages/modules/NeuralNetworks/common/
DCpuExecutor.cpp168 bool setInfoAndAllocateIfNeeded(RunTimeOperandInfo* info, const Shape& shape, int* result) { in setInfoAndAllocateIfNeeded()
244 bool OperationExecutionContext::setOutputShape(uint32_t index, const Shape& shape) { in setOutputShape()
/packages/modules/NeuralNetworks/tools/test_generator/
Dtest_generator.py185 def GetTypeFromString(vt, shape, extraParams=None): argument
193 def GetParsedShape(shape): argument