| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | fully_connected.h | 41 void RearrangeFCWeightsToIOO4I4(const tflite::gpu::Tensor<OHWI, T>& weights, in RearrangeFCWeightsToIOO4I4()
92 void RearrangeFCWeightsToOIO4I4(const tflite::gpu::Tensor<OHWI, T>& weights, in RearrangeFCWeightsToOIO4I4()
157 void FullyConnected::UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights, in UploadWeights()
|
| D | conv_constants.h | 36 const tflite::gpu::Tensor<OHWI, S>& weights, absl::Span<T> dst) { in RearrangeWeightsForConvConstants()
73 const tflite::gpu::Tensor<OHWI, S>& weights, absl::Span<T> dst) { in RearrangeWeightsForConvConstantsDot()
109 void UploadWeightsForConvConstants(const tflite::gpu::Tensor<OHWI, T>& weights, in UploadWeightsForConvConstants()
|
| D | convolution_transposed_thin.h | 71 const tflite::gpu::Tensor<OHWI, T>& weights, in UploadData()
111 const tflite::gpu::Tensor<OHWI, S>& weights, absl::Span<T> dst) { in RearrangeWeightsData()
|
| D | depthwise_conv_3x3.h | 81 const tflite::gpu::Tensor<OHWI, T>& weights, in UploadWeightsAndBiases()
120 const tflite::gpu::Tensor<OHWI, S>& weights, in RearrangeWeightsAndBiasesData()
|
| D | depthwise_conv_3x3_stride_h2.h | 74 const tflite::gpu::Tensor<OHWI, T>& weights, in UploadWeightsAndBiases()
113 const tflite::gpu::Tensor<OHWI, S>& weights, in RearrangeWeightsAndBiasesData()
|
| D | conv_generic.h | 214 void ConvGeneric::UploadData(const tflite::gpu::Tensor<OHWI, T>& weights, in UploadData()
222 const tflite::gpu::Tensor<OHWI, T>& weights) { in UploadDataForWinograd4x4To6x6()
263 void ConvGeneric::UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights) { in UploadWeights()
|
| D | fully_connected.cc | 39 const tflite::gpu::Tensor<OHWI, DataType::INT8>& weights, uint8_t* dst) { in RearrangeFCWeightsToOIO4I4()
198 const tflite::gpu::Tensor<OHWI, DataType::INT8>& weights, float scale, in UploadQuantizedWeights()
|
| D | convolution_transposed.h | 101 const tflite::gpu::Tensor<OHWI, T>& weights, bool weights_are_buffer) { in UploadWeights()
|
| D | depthwise_conv.h | 117 void RearrangeWeightsForDWConv2D(const tflite::gpu::Tensor<OHWI, S>& weights, in RearrangeWeightsForDWConv2D()
147 const tflite::gpu::Tensor<OHWI, T>& weights, bool weights_are_buffer) { in UploadWeightsForDWConv2D()
|
| D | convolution_transposed_3x3_thin.cc | 219 const tflite::gpu::Tensor<OHWI, DataType::FLOAT32>& weights) { in UploadWeights()
|
| D | convolution_transposed_4x4.cc | 372 const tflite::gpu::Tensor<OHWI, DataType::FLOAT32>& weights, in UploadWeights()
|
| D | convolution_transposed_3x3.cc | 382 const tflite::gpu::Tensor<OHWI, DataType::FLOAT32>& weights) { in UploadWeights()
|
| D | conv_weights_converter_test_util.cc | 30 const Tensor<OHWI, DataType::FLOAT32>& weights, in ConvolutionWeightsConverterTest()
|
| D | conv_metal_simd.cc | 377 const tflite::gpu::Tensor<OHWI, DataType::FLOAT32>& weights, in ReorderWeightsForConv()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| D | fc_fc_add.h | 44 void RearrangeFCWeightsToIOO4I4(const tflite::gpu::Tensor<OHWI, T>& weights, in RearrangeFCWeightsToIOO4I4()
71 void RearrangeFCWeightsToOIO4I4(const tflite::gpu::Tensor<OHWI, T>& weights, in RearrangeFCWeightsToOIO4I4()
139 void FCFCAdd::UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights, in UploadWeights()
|
| D | fc_fc_add.cc | 37 const tflite::gpu::Tensor<OHWI, DataType::INT8>& weights, uint8_t* dst) { in RearrangeFCWeightsToOIO4I4()
229 const tflite::gpu::Tensor<OHWI, DataType::INT8>& weights, float scale, in UploadQuantizedWeights()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | weights_conversion.h | 39 const tflite::gpu::Tensor<OHWI, S>& weights, int out_group_size, in RearrangeWeightsToOHWIOGroupI4O4()
113 const tflite::gpu::Tensor<OHWI, S>& weights, int out_group_size, in RearrangeWeightsToOHWIOGroupO4I4()
187 const tflite::gpu::Tensor<OHWI, S>& weights, int out_group_size, in RearrangeWeightsToI4HWIOOGroupO4()
261 const tflite::gpu::Tensor<OHWI, S>& weights, int out_group_size, in RearrangeWeightsToO4HWIOOGroupI4()
335 const tflite::gpu::Tensor<OHWI, S>& weights, in RearrangeWeightsToOICustomSpatialI4O4()
413 const tflite::gpu::Tensor<OHWI, S>& weights, in RearrangeWeightsToOICustomSpatialO4I4()
|
| D | weights_conversion.cc | 61 const tflite::gpu::Tensor<OHWI, DataType::FLOAT32>& weights, in RearrangeWeights()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | winograd_util.cc | 112 const Tensor<OHWI, DataType::FLOAT32>& src_weights, in RearrangeWeightsToWinograd4x4To6x6Weights()
|
| D | shape.h | 55 OHWI = 7, enumerator
203 using OHWI = StrongShape<Layout::OHWI>; variable
|
| D | convert.cc | 97 const Tensor<OHWI, DataType::FLOAT32>& tensor) { in ConvertToPHWO4I4()
107 const Tensor<OHWI, DataType::FLOAT32>& tensor) { in ConvertToPHWO4I4Transposed()
209 const Tensor<OHWI, DataType::FLOAT32>& tensor) { in ConvertToPIOHW4()
|
| /external/ComputeLibrary/src/cpu/kernels/assembly/ |
| D | arm_gemm.hpp | 54 OHWI = 0x100100, enumerator
|
| /external/ComputeLibrary/arm_compute/core/ |
| D | Types.h | 2021 OHWI = 0x100100, enumerator
|