Home
last modified time | relevance | path

Searched refs:convertShapeToDims (Results 1 – 9 of 9) sorted by relevance

/packages/modules/NeuralNetworks/common/operations/
DFullyConnected.cpp70 tflite::reference_ops::FullyConnected(inputData, convertShapeToDims(inputShape), in fullyConnectedFloat32()
71 weightsData, convertShapeToDims(weightsShape), in fullyConnectedFloat32()
72 biasData, convertShapeToDims(biasShape), in fullyConnectedFloat32()
74 outputData, convertShapeToDims(outputShape)); in fullyConnectedFloat32()
77 tflite::optimized_ops::FullyConnected(inputData, convertShapeToDims(inputShape), in fullyConnectedFloat32()
78 weightsData, convertShapeToDims(weightsShape), in fullyConnectedFloat32()
79 biasData, convertShapeToDims(biasShape), in fullyConnectedFloat32()
81 outputData, convertShapeToDims(outputShape)); in fullyConnectedFloat32()
138 tflite::optimized_ops::FullyConnected(inputData, convertShapeToDims(inputShape), inputOffset, in fullyConnectedQuant8()
139 weightsData, convertShapeToDims(weightsShape), in fullyConnectedQuant8()
[all …]
DBroadcast.cpp104 in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), out, \ in addFloat32()
105 convertShapeToDims(shapeOut)) in addFloat32()
113 in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), out, \ in addFloat32()
114 convertShapeToDims(shapeOut)) in addFloat32()
244 in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), out, \ in mulFloat32()
245 convertShapeToDims(shapeOut)) in mulFloat32()
254 tflite::optimized_ops::Mul(in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), in mulFloat32()
256 convertShapeToDims(shapeOut)); in mulFloat32()
320 tflite::optimized_ops::Sub(in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), in subFloat32()
321 out, convertShapeToDims(shapeOut)); in subFloat32()
[all …]
DQuantizedLSTM.cpp412 GetBuffer<const uint8_t>(input_), convertShapeToDims(input_->shape()), in eval()
413 GetBuffer<const uint8_t>(prevOutput_), convertShapeToDims(prevOutput_->shape()), in eval()
414 weights.data(), convertShapeToDims(weightsShape), bias.data(), in eval()
415 convertShapeToDims(biasShape), GetBuffer<const int16_t>(prevCellState_), in eval()
416 convertShapeToDims(prevCellState_->shape()), in eval()
418 GetBuffer<int16_t>(cellStateOut_), convertShapeToDims(cellStateOut_->shape()), in eval()
419 GetBuffer<uint8_t>(output_), convertShapeToDims(output_->shape()), concatTemp.data(), in eval()
420 convertShapeToDims(concatTempShape), activationTemp.data(), in eval()
421 convertShapeToDims(activationTempShape), inputToInputWeights_->zeroPoint, in eval()
DReshape.cpp46 tflite::optimized_ops::DepthToSpace(inputData, convertShapeToDims(inputShape), blockSize, in depthToSpaceGeneric()
47 outputData, convertShapeToDims(outputShape)); in depthToSpaceGeneric()
67 tflite::optimized_ops::SpaceToDepth(inputData, convertShapeToDims(inputShape), blockSize, in spaceToDepthGeneric()
68 outputData, convertShapeToDims(outputShape)); in spaceToDepthGeneric()
204 tflite::optimized_ops::BatchToSpaceND(inputData, convertShapeToDims(inputShape), blockSize, in batchToSpaceGeneric()
206 convertShapeToDims(outputShape)); in batchToSpaceGeneric()
DConcatenation.cpp58 inputDims[i] = convertShapeToDims(inputShapes[i]); in concatenation()
64 inputDimsPtr.data(), num_inputs, outputData, convertShapeToDims(outputShape)); in concatenation()
82 inputDims[i] = convertShapeToDims(inputShapes[i]); in concatenation()
90 convertShapeToDims(outputShape), outputShape.offset, outputShape.scale); in concatenation()
DConv2D.cpp215 inputData, convertShapeToDims(inputShape), filterData, convertShapeToDims(filterShape), in convNhwc()
216 biasData, convertShapeToDims(biasShape), stride_width, stride_height, in convNhwc()
219 convertShapeToDims(outputShape), need_im2colData ? im2colData : nullptr, im2colDim); in convNhwc()
264 tflite::optimized_ops::Conv(inputData, convertShapeToDims(inputShape), inputOffset, filterData, in convNhwc()
265 convertShapeToDims(filterShape), filterOffset, biasData, in convNhwc()
266 convertShapeToDims(biasShape), stride_width, stride_height, in convNhwc()
270 convertShapeToDims(outputShape), in convNhwc()
DTranspose.cpp69 tflite::reference_ops::Transpose(inputData, convertShapeToDims(inputShape), outputData, in transposeGeneric()
70 convertShapeToDims(outputShape), reversed_perm); in transposeGeneric()
DStridedSlice.cpp82 tflite::reference_ops::StridedSlice(inputData, convertShapeToDims(inputShape), beginMask, in compute()
84 convertShapeToDims(outputShape)); in compute()
/packages/modules/NeuralNetworks/common/include/
DCpuOperationUtils.h35 inline tflite::Dims<4> convertShapeToDims(const Shape& shape) { in convertShapeToDims() function