| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/ |
| D | convolution_selector.cc | 37 const Convolution2DAttributes& attr, const BHWC& dst_shape, in SelectConvolutionAdreno()
50 const Convolution2DAttributes& attr, const BHWC& dst_shape, in SelectConvolutionWinogradAdreno()
59 const Convolution2DAttributes& attr, const BHWC& weights_shape, in SelectConvolutionDynamicWeightsAdreno()
70 const Convolution2DAttributes& attr, const BHWC& dst_shape, in SelectConvolutionNVidia()
82 const Convolution2DAttributes& attr, const GpuInfo& gpu_info, in SelectConvolutionPowerVR()
89 const Convolution2DAttributes& attr, const BHWC& dst_shape, in SelectConvolutionMali()
103 const Convolution2DAttributes& attr, const BHWC& dst_shape, in SelectConvolutionWinogradMali()
117 const Convolution2DAttributes& attr, const BHWC& weights_shape, in SelectConvolutionDynamicWeightsMali()
138 const Convolution2DAttributes& attr, const BHWC& dst_shape, in SelectConvolution()
160 const Convolution2DAttributes& attr, const BHWC& dst_shape, in SelectConvolutionForWinograd()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_buffer_1x1.h | 72 const Convolution2DAttributes& attr,
80 const Convolution2DAttributes& attr, const BHWC* shape);
83 const Convolution2DAttributes& attr, const BHWC& weights_shape,
170 const Convolution2DAttributes& attr);
174 const Convolution2DAttributes& attr);
178 const Convolution2DAttributes& attr,
188 const Convolution2DAttributes& attr, const BHWC& weights_shape,
193 const Convolution2DAttributes& attr, const BHWC* shape = nullptr);
|
| D | conv_metal.h | 84 const Convolution2DAttributes& attr, const GpuInfo& gpu_info);
88 const Convolution2DAttributes& attr, const GpuInfo& gpu_info);
97 const Convolution2DAttributes& attr,
102 const Convolution2DAttributes& attr, const GpuInfo& gpu_info);
|
| D | conv_powervr.h | 111 const Convolution2DAttributes& attr, const GpuInfo& gpu_info,
114 const Convolution2DAttributes& attr, const BHWC& weights_shape,
144 const Convolution2DAttributes& attr,
154 const Convolution2DAttributes& attr, const BHWC& weights_shape,
159 const Convolution2DAttributes& attr, const BHWC* dst_shape);
168 const Convolution2DAttributes& attr,
172 const Convolution2DAttributes& attr,
181 const Convolution2DAttributes& attr,
361 const Convolution2DAttributes& attr,
371 const Convolution2DAttributes& attr,
[all …]
|
| D | conv_powervr_test_util.cc | 33 Convolution2DAttributes attr; in ConvPowerVR1x1SimpleWeightsTest()
70 Convolution2DAttributes attr; in ConvPowerVR1x1Test()
107 Convolution2DAttributes attr; in ConvPowerVRSimpleWeightsTest()
143 Convolution2DAttributes attr; in ConvPowerVRTest()
|
| D | conv_buffer_1x1.cc | 322 const Convolution2DAttributes& attr) { in IsConvBuffer1x1Supported()
334 const Convolution2DAttributes& attr) { in IsConvBuffer1x1Supported()
346 const Convolution2DAttributes& attr, in CreateConvBuffer1x1()
384 const Convolution2DAttributes& attr, const BHWC* shape) { in CreateConvBuffer1x1Wino4x4To6x6()
404 const Convolution2DAttributes& attr, const BHWC& weights_shape, in CreateConvBuffer1x1DynamicWeights()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | convolution_selector.h | 32 const Convolution2DAttributes& attr, const BHWC& dst_shape,
36 const Convolution2DAttributes& attr, const BHWC& dst_shape,
40 const Convolution2DAttributes& attr, const BHWC& weights_shape,
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| D | depthwise_conv_plus_1x1_conv.h | 37 const Convolution2DAttributes& conv_attr);
42 const Convolution2DAttributes& conv_attr);
|
| D | depthwise_conv_plus_1x1_conv.cc | 28 const Convolution2DAttributes& conv_attr, in UploadWeights()
236 const Convolution2DAttributes& conv_attr) { in IsDepthwiseConvPlus1x1ConvSupported()
256 const Convolution2DAttributes& conv_attr) { in CreateDepthwiseConvPlus1x1Conv()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | winograd_util_test.cc | 25 Convolution2DAttributes attr; in TEST()
35 Convolution2DAttributes attr; in TEST()
|
| D | operations.h | 253 struct Convolution2DAttributes { struct
274 const Convolution2DAttributes& attr); argument
284 const Convolution2DAttributes& attr);
325 struct DepthwiseConvolution2DAttributes : public Convolution2DAttributes {};
|
| D | winograd_util.h | 42 bool IsSuitableForWinograd4x4To6x6(const Convolution2DAttributes& attr);
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| D | fuse_mul_to_conv.h | 41 Convolution2DAttributes* attr);
67 Convolution2DAttributes* attr);
|
| D | make_fully_connected_test.cc | 40 Convolution2DAttributes attr0; in TEST()
48 Convolution2DAttributes attr1; in TEST()
56 Convolution2DAttributes attr2; in TEST()
|
| D | fuse_add_to_conv.cc | 75 Convolution2DAttributes* conv_attr = in ApplyToNodesSequence()
76 absl::any_cast<Convolution2DAttributes>( in ApplyToNodesSequence()
118 Convolution2DAttributes* attr) { in FuseConvolution2DWithAdd()
|
| D | make_fully_connected.cc | 34 bool IsConvEquivalentToFullyConnected(const Convolution2DAttributes& attr) { in IsConvEquivalentToFullyConnected()
59 const auto& conv_attr = absl::any_cast<const Convolution2DAttributes&>( in ApplyToNode()
|
| D | fuse_mul_to_conv.cc | 66 Convolution2DAttributes* conv_attr = in ApplyToNodesSequence()
67 absl::any_cast<Convolution2DAttributes>( in ApplyToNodesSequence()
130 Convolution2DAttributes* conv_attr = in ApplyToNodesSequence()
131 absl::any_cast<Convolution2DAttributes>( in ApplyToNodesSequence()
177 Convolution2DAttributes* attr) { in FuseConvolution2DWithMultiply()
256 Convolution2DAttributes* attr) { in FuseMultiplyWithConvolution2D()
|
| D | merge_padding_with_test.cc | 53 Convolution2DAttributes conv_attr; in TEST()
68 absl::any_cast<Convolution2DAttributes>(conv_node->operation.attributes); in TEST()
98 Convolution2DAttributes conv_attr; in TEST()
113 absl::any_cast<Convolution2DAttributes>(conv_node->operation.attributes); in TEST()
|
| D | fuse_mul_to_conv_test.cc | 45 Convolution2DAttributes conv_attr; in TEST()
102 Convolution2DAttributes conv_attr; in TEST()
143 Convolution2DAttributes attr; in TEST()
227 Convolution2DAttributes attr; in TEST()
|
| D | fuse_add_to_conv.h | 36 Convolution2DAttributes* attr);
|
| D | fuse_add_to_conv_test.cc | 45 Convolution2DAttributes conv_attr; in TEST()
92 Convolution2DAttributes attr; in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | conv_test.cc | 39 Convolution2DAttributes attr; in TEST()
77 Convolution2DAttributes attr; in TEST()
114 Convolution2DAttributes attr; in TEST()
151 Convolution2DAttributes attr; in TEST()
188 Convolution2DAttributes attr; in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | conv_buffer_1x1_test.cc | 39 Convolution2DAttributes attr; in TEST_F()
75 Convolution2DAttributes attr; in TEST_F()
|
| D | conv_constants_test.cc | 39 Convolution2DAttributes attr; in TEST_F()
75 Convolution2DAttributes attr; in TEST_F()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/ |
| D | conv_test.mm | 48 Convolution2DAttributes attr;
84 Convolution2DAttributes attr;
120 Convolution2DAttributes attr;
156 Convolution2DAttributes attr;
192 Convolution2DAttributes attr;
226 Convolution2DAttributes attr;
|