| /frameworks/ml/nn/common/operations/ |
| D | Broadcast.cpp | 92 in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), out, \ in addFloat32()
93 convertShapeToDims(shapeOut)) in addFloat32()
101 in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), out, \ in addFloat32()
102 convertShapeToDims(shapeOut)) in addFloat32()
159 left_shift, in1, convertShapeToDims(shape1), input1_offset, input1_multiplier, \ in addQuant8()
160 input1_shift, in2, convertShapeToDims(shape2), input2_offset, input2_multiplier, \ in addQuant8()
162 output_activation_max, out, convertShapeToDims(shapeOut)) in addQuant8()
170 left_shift, in1, convertShapeToDims(shape1), input1_offset, input1_multiplier, \ in addQuant8()
171 input1_shift, in2, convertShapeToDims(shape2), input2_offset, input2_multiplier, \ in addQuant8()
173 output_activation_max, out, convertShapeToDims(shapeOut)) in addQuant8()
[all …]
|
| D | StridedSlice.cpp | 64 reinterpret_cast<const float*>(inputData), convertShapeToDims(inputShape), in stridedSliceGeneric()
66 reinterpret_cast<float*>(outputData), convertShapeToDims(outputShape)); in stridedSliceGeneric()
70 reinterpret_cast<const _Float16*>(inputData), convertShapeToDims(inputShape), in stridedSliceGeneric()
72 reinterpret_cast<_Float16*>(outputData), convertShapeToDims(outputShape)); in stridedSliceGeneric()
76 reinterpret_cast<const uint8_t*>(inputData), convertShapeToDims(inputShape), in stridedSliceGeneric()
78 reinterpret_cast<uint8_t*>(outputData), convertShapeToDims(outputShape)); in stridedSliceGeneric()
|
| D | FullyConnected.cpp | 61 tflite::reference_ops::FullyConnected(inputData, convertShapeToDims(inputShape), in fullyConnectedFloat32()
62 weightsData, convertShapeToDims(weightsShape), in fullyConnectedFloat32()
63 biasData, convertShapeToDims(biasShape), in fullyConnectedFloat32()
65 outputData, convertShapeToDims(outputShape)); in fullyConnectedFloat32()
68 tflite::optimized_ops::FullyConnected(inputData, convertShapeToDims(inputShape), in fullyConnectedFloat32()
69 weightsData, convertShapeToDims(weightsShape), in fullyConnectedFloat32()
70 biasData, convertShapeToDims(biasShape), in fullyConnectedFloat32()
72 outputData, convertShapeToDims(outputShape)); in fullyConnectedFloat32()
129 tflite::optimized_ops::FullyConnected(inputData, convertShapeToDims(inputShape), inputOffset, in fullyConnectedQuant8()
130 weightsData, convertShapeToDims(weightsShape), in fullyConnectedQuant8()
[all …]
|
| D | Reshape.cpp | 44 tflite::optimized_ops::DepthToSpace(inputData, convertShapeToDims(inputShape), blockSize, in depthToSpaceGeneric()
45 outputData, convertShapeToDims(outputShape)); in depthToSpaceGeneric()
62 tflite::optimized_ops::SpaceToDepth(inputData, convertShapeToDims(inputShape), blockSize, in spaceToDepthGeneric()
63 outputData, convertShapeToDims(outputShape)); in spaceToDepthGeneric()
193 tflite::optimized_ops::BatchToSpaceND(inputData, convertShapeToDims(inputShape), blockSize, in batchToSpaceGeneric()
195 convertShapeToDims(outputShape)); in batchToSpaceGeneric()
|
| D | QuantizedLSTM.cpp | 410 GetBuffer<const uint8_t>(input_), convertShapeToDims(input_->shape()), in eval()
411 GetBuffer<const uint8_t>(prevOutput_), convertShapeToDims(prevOutput_->shape()), in eval()
412 weights.data(), convertShapeToDims(weightsShape), bias.data(), in eval()
413 convertShapeToDims(biasShape), GetBuffer<const int16_t>(prevCellState_), in eval()
414 convertShapeToDims(prevCellState_->shape()), in eval()
416 GetBuffer<int16_t>(cellStateOut_), convertShapeToDims(cellStateOut_->shape()), in eval()
417 GetBuffer<uint8_t>(output_), convertShapeToDims(output_->shape()), concatTemp.data(), in eval()
418 convertShapeToDims(concatTempShape), activationTemp.data(), in eval()
419 convertShapeToDims(activationTempShape), inputToInputWeights_->zeroPoint, in eval()
|
| D | Concatenation.cpp | 45 inputDims[i] = convertShapeToDims(inputShapes[i]); in concatenation()
51 inputDimsPtr.data(), num_inputs, outputData, convertShapeToDims(outputShape)); in concatenation()
69 inputDims[i] = convertShapeToDims(inputShapes[i]); in concatenation()
77 convertShapeToDims(outputShape), outputShape.offset, outputShape.scale); in concatenation()
|
| D | Conv2D.cpp | 180 tflite::optimized_ops::Conv(inputData, convertShapeToDims(inputShape), filterData, in convNhwc()
181 convertShapeToDims(filterShape), biasData, in convNhwc()
182 convertShapeToDims(biasShape), stride_width, stride_height, in convNhwc()
185 outputData, convertShapeToDims(outputShape), im2colData, im2colDim); in convNhwc()
227 inputData, convertShapeToDims(inputShape), inputOffset, filterData, in convNhwc()
228 convertShapeToDims(filterShape), filterOffset, biasData, convertShapeToDims(biasShape), in convNhwc()
232 convertShapeToDims(outputShape), im2colData, im2colDim, &gemm_context); in convNhwc()
|
| D | Transpose.cpp | 62 tflite::reference_ops::Transpose(inputData, convertShapeToDims(inputShape), outputData, in transposeGeneric()
63 convertShapeToDims(outputShape), reversed_perm); in transposeGeneric()
|
| D | SimpleMath.cpp | 45 tflite::Dims<4> dim = convertShapeToDims(shape); in floorFloat32()
|
| /frameworks/ml/nn/common/include/ |
| D | CpuOperationUtils.h | 33 inline tflite::Dims<4> convertShapeToDims(const Shape& shape) { in convertShapeToDims() function
|