| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_buffer_1x1.h | 107 void ConvBuffer1x1::UploadData(const tflite::gpu::Tensor<OHWI, T>& weights, in UploadData()
115 const tflite::gpu::Tensor<OHWI, T>& weights) { in UploadDataForWinograd4x4To6x6()
126 void ConvBuffer1x1::UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights) { in UploadWeights()
|
| D | fully_connected.h | 40 void RearrangeFCWeightsToIOO4I4(const tflite::gpu::Tensor<OHWI, T>& weights, in RearrangeFCWeightsToIOO4I4()
91 void RearrangeFCWeightsToOIO4I4(const tflite::gpu::Tensor<OHWI, T>& weights, in RearrangeFCWeightsToOIO4I4()
148 void FullyConnected::UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights, in UploadWeights()
|
| D | conv_constants.h | 35 const tflite::gpu::Tensor<OHWI, S>& weights, absl::Span<T> dst) { in RearrangeWeightsForConvConstants()
72 const tflite::gpu::Tensor<OHWI, S>& weights, absl::Span<T> dst) { in RearrangeWeightsForConvConstantsDot()
108 void UploadWeightsForConvConstants(const tflite::gpu::Tensor<OHWI, T>& weights, in UploadWeightsForConvConstants()
|
| D | convolution_transposed_thin.h | 68 const tflite::gpu::Tensor<OHWI, T>& weights, in UploadData()
108 const tflite::gpu::Tensor<OHWI, S>& weights, absl::Span<T> dst) { in RearrangeWeightsData()
|
| D | depthwise_conv_3x3.h | 78 const tflite::gpu::Tensor<OHWI, T>& weights, in UploadWeightsAndBiases()
118 const tflite::gpu::Tensor<OHWI, S>& weights, in RearrangeWeightsAndBiasesData()
|
| D | depthwise_conv_3x3_stride_h2.h | 72 const tflite::gpu::Tensor<OHWI, T>& weights, in UploadWeightsAndBiases()
112 const tflite::gpu::Tensor<OHWI, S>& weights, in RearrangeWeightsAndBiasesData()
|
| D | conv_powervr.h | 206 void ConvPowerVR::UploadData(const tflite::gpu::Tensor<OHWI, T>& weights, in UploadData()
214 const tflite::gpu::Tensor<OHWI, T>& weights) { in UploadDataForWinograd4x4To6x6()
255 void ConvPowerVR::UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights) { in UploadWeights()
|
| D | depthwise_conv.h | 37 void RearrangeWeightsForDWConv2D(const tflite::gpu::Tensor<OHWI, S>& weights, in RearrangeWeightsForDWConv2D()
66 void UploadWeightsForDWConv2D(const tflite::gpu::Tensor<OHWI, T>& weights, in UploadWeightsForDWConv2D()
|
| D | convolution_transposed.h | 99 const tflite::gpu::Tensor<OHWI, T>& weights, bool weights_are_buffer) { in UploadWeights()
|
| D | convolution_transposed_3x3_thin.cc | 204 const tflite::gpu::Tensor<OHWI, DataType::FLOAT32>& weights) { in UploadWeights()
|
| D | convolution_transposed_4x4.cc | 392 const tflite::gpu::Tensor<OHWI, DataType::FLOAT32>& weights, in UploadWeights()
|
| D | convolution_transposed_3x3.cc | 388 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.cc | 593 const tflite::gpu::Tensor<OHWI, DataType::FLOAT32>& weights, in ReorderWeightsForConv()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| D | fc_fc_add.h | 40 void RearrangeFCWeightsToIOO4I4(const tflite::gpu::Tensor<OHWI, T>& weights, in RearrangeFCWeightsToIOO4I4()
67 void RearrangeFCWeightsToOIO4I4(const tflite::gpu::Tensor<OHWI, T>& weights, in RearrangeFCWeightsToOIO4I4()
125 void FCFCAdd::UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights, in UploadWeights()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | weights_conversion.h | 37 const tflite::gpu::Tensor<OHWI, S>& weights, int out_group_size, in RearrangeWeightsToOHWIOGroupI4O4()
73 const tflite::gpu::Tensor<OHWI, S>& weights, int out_group_size, in RearrangeWeightsToOHWIOGroupO4I4()
147 const tflite::gpu::Tensor<OHWI, S>& weights, int out_group_size, in RearrangeWeightsToI4HWIOOGroupO4()
183 const tflite::gpu::Tensor<OHWI, S>& weights, int out_group_size, in RearrangeWeightsToO4HWIOOGroupI4()
257 const tflite::gpu::Tensor<OHWI, S>& weights, in RearrangeWeightsToOICustomSpatialI4O4()
293 const tflite::gpu::Tensor<OHWI, S>& weights, in RearrangeWeightsToOICustomSpatialO4I4()
|
| D | weights_conversion.cc | 40 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
202 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()
|