| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | convert.h | 55 uint32_t GetElementsSizeForPIOHW4(const OHWI& shape);
59 absl::Status ConvertToPIOHW4(absl::Span<const float> in, const OHWI& shape,
64 const Tensor<OHWI, DataType::FLOAT32>& tensor);
67 uint32_t GetElementsSizeForPHWO4I4(const OHWI& shape);
71 const Tensor<OHWI, DataType::FLOAT32>& tensor);
75 const Tensor<OHWI, DataType::FLOAT32>& tensor);
79 uint3 Get3DSizeForPHWO4I4(const OHWI& shape);
|
| D | winograd_util.h | 39 const Tensor<OHWI, DataType::FLOAT32>& src_weights,
40 Tensor<OHWI, DataType::FLOAT32>* dst_weights);
|
| D | winograd_util_test.cc | 30 attr.weights.shape = OHWI(1, 3, 3, 1); in TEST()
40 attr.weights.shape = OHWI(1, 2, 3, 1); in TEST()
|
| D | convert.cc | 43 absl::Status ConvertToPHWO4I4(absl::Span<const float> in, const OHWI& shape, in ConvertToPHWO4I4()
86 uint32_t GetElementsSizeForPHWO4I4(const OHWI& shape) { in GetElementsSizeForPHWO4I4()
97 const Tensor<OHWI, DataType::FLOAT32>& tensor) { in ConvertToPHWO4I4() argument
107 const Tensor<OHWI, DataType::FLOAT32>& tensor) { in ConvertToPHWO4I4Transposed() argument
116 uint3 Get3DSizeForPHWO4I4(const OHWI& shape) { in Get3DSizeForPHWO4I4()
171 uint32_t GetElementsSizeForPIOHW4(const OHWI& shape) { in GetElementsSizeForPIOHW4()
176 absl::Status ConvertToPIOHW4(absl::Span<const float> in, const OHWI& shape, in ConvertToPIOHW4()
209 const Tensor<OHWI, DataType::FLOAT32>& tensor) { in ConvertToPIOHW4() argument
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | transpose_conv_test.cc | 46 Tensor<OHWI, DataType::FLOAT32> weights; in TEST()
47 weights.shape = OHWI(2, 2, 1, 1); in TEST()
85 Tensor<OHWI, DataType::FLOAT32> weights; in TEST()
86 weights.shape = OHWI(1, 2, 2, 1); in TEST()
126 Tensor<OHWI, DataType::FLOAT32> weights; in TEST()
127 weights.shape = OHWI(1, 3, 3, 1); in TEST()
165 Tensor<OHWI, DataType::FLOAT32> weights; in TEST()
166 weights.shape = OHWI(2, 1, 1, 2); in TEST()
203 Tensor<OHWI, DataType::FLOAT32> weights; in TEST()
204 weights.shape = OHWI(1, 1, 1, 1); in TEST()
|
| D | conv_test.cc | 46 Tensor<OHWI, DataType::FLOAT32> weights; in TEST()
47 weights.shape = OHWI(2, 2, 1, 1); in TEST()
84 Tensor<OHWI, DataType::FLOAT32> weights; in TEST()
85 weights.shape = OHWI(1, 2, 2, 1); in TEST()
121 Tensor<OHWI, DataType::FLOAT32> weights; in TEST()
122 weights.shape = OHWI(1, 3, 3, 1); in TEST()
158 Tensor<OHWI, DataType::FLOAT32> weights; in TEST()
159 weights.shape = OHWI(2, 1, 1, 2); in TEST()
195 Tensor<OHWI, DataType::FLOAT32> weights; in TEST()
196 weights.shape = OHWI(1, 1, 1, 1); in TEST()
|
| D | depthwise_conv_test.cc | 46 Tensor<OHWI, DataType::FLOAT32> weights; in TEST()
47 weights.shape = OHWI(2, 1, 1, 2); in TEST()
84 Tensor<OHWI, DataType::FLOAT32> weights; in TEST()
85 weights.shape = OHWI(2, 1, 1, 1); in TEST()
123 Tensor<OHWI, DataType::FLOAT32> weights; in TEST()
124 weights.shape = OHWI(2, 2, 2, 1); in TEST()
|
| D | fully_connected_test.cc | 47 Tensor<OHWI, DataType::FLOAT32> weights; in TEST()
48 weights.shape = OHWI(4, 1, 1, 2); in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_weights_converter_test_util.cc | 30 const Tensor<OHWI, DataType::FLOAT32>& weights, in ConvolutionWeightsConverterTest() argument
110 auto weights_shape = OHWI(kDstChannels, 1, 1, kSrcChannels); in ConverterToConvWeights1x1OutX4Test()
114 Tensor<OHWI, DataType::FLOAT32> weights; in ConverterToConvWeights1x1OutX4Test()
144 auto weights_shape = OHWI(kDstChannels, 1, 1, kSrcChannels); in ConverterToConvWeights1x1OutX4UnalignedTest()
148 Tensor<OHWI, DataType::FLOAT32> weights; in ConverterToConvWeights1x1OutX4UnalignedTest()
177 auto weights_shape = OHWI(kDstChannels, 1, 1, kSrcChannels); in ConverterToConvWeights1x1OutX2Test()
181 Tensor<OHWI, DataType::FLOAT32> weights; in ConverterToConvWeights1x1OutX2Test()
210 auto weights_shape = OHWI(kDstChannels, 3, 4, kSrcChannels); in ConverterToConvWeightsOutX2Test()
214 Tensor<OHWI, DataType::FLOAT32> weights; in ConverterToConvWeightsOutX2Test()
244 auto weights_shape = OHWI(kDstChannels, 4, 4, kSrcChannels); in ConverterToConvTransposedWeights4x4Test()
[all …]
|
| D | conv_buffer_1x1.h | 87 void UploadData(const tflite::gpu::Tensor<OHWI, T>& weights,
91 const tflite::gpu::Tensor<OHWI, T>& weights);
94 void UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights);
107 void ConvBuffer1x1::UploadData(const tflite::gpu::Tensor<OHWI, T>& weights, in UploadData() argument
115 const tflite::gpu::Tensor<OHWI, T>& weights) { in UploadDataForWinograd4x4To6x6() argument
116 tflite::gpu::Tensor<OHWI, T> wino_weights; in UploadDataForWinograd4x4To6x6()
126 void ConvBuffer1x1::UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights) { in UploadWeights() argument
|
| D | depthwise_conv_3x3_stride_h2.h | 59 void UploadWeightsAndBiases(const tflite::gpu::Tensor<OHWI, T>& weights,
64 const tflite::gpu::Tensor<OHWI, S>& weights,
72 const tflite::gpu::Tensor<OHWI, T>& weights, in UploadWeightsAndBiases() argument
112 const tflite::gpu::Tensor<OHWI, S>& weights, in RearrangeWeightsAndBiasesData() argument
|
| D | convolution_transposed_thin.h | 55 void UploadData(const tflite::gpu::Tensor<OHWI, T>& weights,
59 void RearrangeWeightsData(const tflite::gpu::Tensor<OHWI, S>& weights,
68 const tflite::gpu::Tensor<OHWI, T>& weights, in UploadData() argument
108 const tflite::gpu::Tensor<OHWI, S>& weights, absl::Span<T> dst) { in RearrangeWeightsData() argument
|
| D | depthwise_conv_3x3.h | 56 void UploadWeightsAndBiases(const tflite::gpu::Tensor<OHWI, T>& weights,
66 const tflite::gpu::Tensor<OHWI, S>& weights,
78 const tflite::gpu::Tensor<OHWI, T>& weights, in UploadWeightsAndBiases() argument
118 const tflite::gpu::Tensor<OHWI, S>& weights, in RearrangeWeightsAndBiasesData() argument
|
| D | fully_connected.h | 40 void RearrangeFCWeightsToIOO4I4(const tflite::gpu::Tensor<OHWI, T>& weights, in RearrangeFCWeightsToIOO4I4() argument
91 void RearrangeFCWeightsToOIO4I4(const tflite::gpu::Tensor<OHWI, T>& weights, in RearrangeFCWeightsToOIO4I4() argument
140 void UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights,
148 void FullyConnected::UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights, in UploadWeights() argument
|
| D | conv_powervr.h | 127 void UploadData(const tflite::gpu::Tensor<OHWI, T>& weights,
131 const tflite::gpu::Tensor<OHWI, T>& weights);
134 void UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights);
206 void ConvPowerVR::UploadData(const tflite::gpu::Tensor<OHWI, T>& weights, in UploadData() argument
214 const tflite::gpu::Tensor<OHWI, T>& weights) { in UploadDataForWinograd4x4To6x6() argument
215 tflite::gpu::Tensor<OHWI, T> wino_weights; in UploadDataForWinograd4x4To6x6()
255 void ConvPowerVR::UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights) { in UploadWeights() argument
|
| D | conv_powervr_test_util.cc | 38 attr.weights.shape = OHWI(2, 1, 1, 2); in ConvPowerVR1x1SimpleWeightsTest()
75 attr.weights.shape = OHWI(2, 1, 1, 2); in ConvPowerVR1x1Test()
112 attr.weights.shape = OHWI(1, 2, 2, 2); in ConvPowerVRSimpleWeightsTest()
148 attr.weights.shape = OHWI(2, 2, 2, 2); in ConvPowerVRTest()
|
| D | depthwise_conv_test_util.cc | 38 attr.weights.shape = OHWI(1, 3, 1, 2); in DepthwiseConvSimpleWeightsTest()
75 attr.weights.shape = OHWI(1, 3, 1, 2); in DepthwiseConvNoMultiplierTest()
112 attr.weights.shape = OHWI(2, 3, 1, 2); in DepthwiseConvMultiplier2Test()
|
| D | fully_connected_test_util.cc | 34 attr.weights.shape = OHWI(2, 1, 1, 4); in FullyConnectedTest()
67 attr.weights.shape = OHWI(12, 1, 1, 8); in FullyConnectedLargeTest()
118 attr.weights.shape = OHWI(1024, 1, 1, kInputSize); in FullyConnectedExtraLargeTest()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | fully_connected_test.cc | 54 tflite::gpu::Tensor<OHWI, DataType::FLOAT32> weights; in TEST_F()
55 weights.shape = OHWI(8, 1, 1, 8); in TEST_F()
88 tflite::gpu::Tensor<OHWI, DataType::FLOAT32> weights; in TEST_F()
89 weights.shape = OHWI(9, 1, 1, 7); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| D | fuse_mul_to_conv_test.cc | 50 conv_attr.weights.shape = OHWI(16, 3, 2, 8); in TEST()
107 conv_attr.weights.shape = OHWI(16, 3, 2, 8); in TEST()
144 attr.weights.shape = OHWI(2, 1, 2, 2); in TEST()
165 attr.weights.shape = OHWI(2, 1, 2, 2); in TEST()
187 attr.weights.shape = OHWI(2, 1, 2, 2); in TEST()
208 attr.weights.shape = OHWI(2, 1, 1, 2); in TEST()
228 attr.weights.shape = OHWI(2, 1, 2, 2); in TEST()
249 attr.weights.shape = OHWI(2, 1, 2, 2); in TEST()
271 attr.weights.shape = OHWI(2, 1, 2, 2); in TEST()
292 attr.weights.shape = OHWI(2, 1, 1, 2); in TEST()
|
| D | fuse_add_to_conv_test.cc | 50 conv_attr.weights.shape = OHWI(16, 3, 2, 8); in TEST()
93 attr.weights.shape = OHWI(2, 1, 2, 2); in TEST()
114 attr.weights.shape = OHWI(2, 1, 2, 2); in TEST()
136 attr.weights.shape = OHWI(2, 1, 2, 2); in TEST()
157 attr.weights.shape = OHWI(2, 1, 1, 2); in TEST()
|
| D | make_fully_connected_test.cc | 45 attr0.weights.shape = OHWI(16, 1, 1, 8); in TEST()
53 attr1.weights.shape = OHWI(16, 4, 4, 16); in TEST()
61 attr2.weights.shape = OHWI(32, 1, 1, 16); in TEST()
107 EXPECT_EQ(OHWI(32, 1, 1, 16), fc_attr.weights.shape); in TEST()
|
| /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() argument
73 const tflite::gpu::Tensor<OHWI, S>& weights, int out_group_size, in RearrangeWeightsToOHWIOGroupO4I4() argument
147 const tflite::gpu::Tensor<OHWI, S>& weights, int out_group_size, in RearrangeWeightsToI4HWIOOGroupO4() argument
183 const tflite::gpu::Tensor<OHWI, S>& weights, int out_group_size, in RearrangeWeightsToO4HWIOOGroupI4() argument
257 const tflite::gpu::Tensor<OHWI, S>& weights, in RearrangeWeightsToOICustomSpatialI4O4() argument
293 const tflite::gpu::Tensor<OHWI, S>& weights, in RearrangeWeightsToOICustomSpatialO4I4() argument
328 const OHWI& shape);
331 const tflite::gpu::Tensor<OHWI, DataType::FLOAT32>& weights,
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/workgroups/ |
| D | ideal_workgroup_picker.h | 33 OHWI workload);
40 uint3 default_wg, OHWI workload);
|
| /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() argument
67 void RearrangeFCWeightsToOIO4I4(const tflite::gpu::Tensor<OHWI, T>& weights, in RearrangeFCWeightsToOIO4I4() argument
117 void UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights,
125 void FCFCAdd::UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights, in UploadWeights() argument
|