Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DFullyConnected.cpp69 tflite::reference_ops::FullyConnected(inputData, convertShapeToDims(inputShape), in fullyConnectedFloat32()
70 weightsData, convertShapeToDims(weightsShape), in fullyConnectedFloat32()
71 biasData, convertShapeToDims(biasShape), in fullyConnectedFloat32()
73 outputData, convertShapeToDims(outputShape)); in fullyConnectedFloat32()
76 tflite::optimized_ops::FullyConnected(inputData, convertShapeToDims(inputShape), in fullyConnectedFloat32()
77 weightsData, convertShapeToDims(weightsShape), in fullyConnectedFloat32()
78 biasData, convertShapeToDims(biasShape), in fullyConnectedFloat32()
80 outputData, convertShapeToDims(outputShape)); in fullyConnectedFloat32()
137 tflite::optimized_ops::FullyConnected(inputData, convertShapeToDims(inputShape), inputOffset, in fullyConnectedQuant8()
138 weightsData, convertShapeToDims(weightsShape), in fullyConnectedQuant8()
[all …]
DBroadcast.cpp101 in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), out, \ in addFloat32()
102 convertShapeToDims(shapeOut)) in addFloat32()
110 in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), out, \ in addFloat32()
111 convertShapeToDims(shapeOut)) in addFloat32()
241 in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), out, \ in mulFloat32()
242 convertShapeToDims(shapeOut)) in mulFloat32()
251 tflite::optimized_ops::Mul(in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), in mulFloat32()
253 convertShapeToDims(shapeOut)); in mulFloat32()
317 tflite::optimized_ops::Sub(in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), in subFloat32()
318 out, convertShapeToDims(shapeOut)); in subFloat32()
[all …]
DQuantizedLSTM.cpp415 GetBuffer<const uint8_t>(input_), convertShapeToDims(input_->shape()), in eval()
416 GetBuffer<const uint8_t>(prevOutput_), convertShapeToDims(prevOutput_->shape()), in eval()
417 weights.data(), convertShapeToDims(weightsShape), bias.data(), in eval()
418 convertShapeToDims(biasShape), GetBuffer<const int16_t>(prevCellState_), in eval()
419 convertShapeToDims(prevCellState_->shape()), in eval()
421 GetBuffer<int16_t>(cellStateOut_), convertShapeToDims(cellStateOut_->shape()), in eval()
422 GetBuffer<uint8_t>(output_), convertShapeToDims(output_->shape()), concatTemp.data(), in eval()
423 convertShapeToDims(concatTempShape), activationTemp.data(), in eval()
424 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.cpp56 inputDims[i] = convertShapeToDims(inputShapes[i]); in concatenation()
62 inputDimsPtr.data(), num_inputs, outputData, convertShapeToDims(outputShape)); in concatenation()
80 inputDims[i] = convertShapeToDims(inputShapes[i]); in concatenation()
88 convertShapeToDims(outputShape), outputShape.offset, outputShape.scale); in concatenation()
DTranspose.cpp69 tflite::reference_ops::Transpose(inputData, convertShapeToDims(inputShape), outputData, in transposeGeneric()
70 convertShapeToDims(outputShape), reversed_perm); in transposeGeneric()
DConv2D.cpp214 inputData, convertShapeToDims(inputShape), filterData, convertShapeToDims(filterShape), in convNhwc()
215 biasData, convertShapeToDims(biasShape), stride_width, stride_height, in convNhwc()
218 convertShapeToDims(outputShape), need_im2colData ? im2colData : nullptr, im2colDim); in convNhwc()
263 tflite::optimized_ops::Conv(inputData, convertShapeToDims(inputShape), inputOffset, filterData, in convNhwc()
264 convertShapeToDims(filterShape), filterOffset, biasData, in convNhwc()
265 convertShapeToDims(biasShape), stride_width, stride_height, in convNhwc()
269 convertShapeToDims(outputShape), in convNhwc()
DStridedSlice.cpp81 tflite::reference_ops::StridedSlice(inputData, convertShapeToDims(inputShape), beginMask, in compute()
83 convertShapeToDims(outputShape)); in compute()
/frameworks/ml/nn/common/include/
DCpuOperationUtils.h34 inline tflite::Dims<4> convertShapeToDims(const Shape& shape) { in convertShapeToDims() function