Searched refs:inputData_float32 (Results 1 – 6 of 6) sorted by relevance
/frameworks/ml/nn/common/operations/ |
D | Softmax.cpp | 102 std::vector<float> inputData_float32(getNumberOfElements(inputShape)); in softmaxFloat16() local 103 convertFloat16ToFloat32(inputData, &inputData_float32); in softmaxFloat16() 106 softmaxFloat32(inputData_float32.data(), inputShape, beta, axis, outputData_float32.data(), in softmaxFloat16()
|
D | ResizeImageOps.cpp | 140 std::vector<float> inputData_float32(getNumberOfElements(inputShape)); in resizeImageOpNhwc() local 141 convertFloat16ToFloat32(inputData, &inputData_float32); in resizeImageOpNhwc() 143 NN_RET_CHECK(resizeImageOpNhwc(opType, inputData_float32.data(), inputShape, alignCorners, in resizeImageOpNhwc()
|
D | GroupedConv2D.cpp | 293 std::vector<float> inputData_float32(getNumberOfElements(inputShape)); in groupedConvFloat16() local 298 convertFloat16ToFloat32(inputData, &inputData_float32); in groupedConvFloat16() 302 groupedConvFloat32(inputData_float32.data(), inputShape, filterData_float32.data(), filterShape, in groupedConvFloat16()
|
D | Pooling.cpp | 242 std::vector<float> inputData_float32(getNumberOfElements(inputShape)); in maxPoolNhwc() local 245 convertFloat16ToFloat32(inputData, &inputData_float32); in maxPoolNhwc() 246 maxPoolNhwc(inputData_float32.data(), inputShape, param, outputData_float32.data(), in maxPoolNhwc()
|
D | TransposeConv2D.cpp | 282 std::vector<float> inputData_float32(getNumberOfElements(inputShape)); in transposeConvNhwc() local 287 convertFloat16ToFloat32(inputData, &inputData_float32); in transposeConvNhwc() 291 transposeConvNhwc(inputData_float32.data(), inputShape, filterData_float32.data(), filterShape, in transposeConvNhwc()
|
D | Conv2D.cpp | 313 std::vector<float> inputData_float32(getNumberOfElements(inputShape)); in convNhwc() local 318 convertFloat16ToFloat32(inputData, &inputData_float32); in convNhwc() 322 convNhwc(inputData_float32.data(), inputShape, filterData_float32.data(), filterShape, in convNhwc()
|