Home
last modified time | relevance | path

Searched refs:ConvBuffer1x1 (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_buffer_1x1.h35 class ConvBuffer1x1 : public GPUOperation {
37 ConvBuffer1x1() = default;
40 ConvBuffer1x1(ConvBuffer1x1&& operation);
41 ConvBuffer1x1& operator=(ConvBuffer1x1&& operation);
42 ConvBuffer1x1(const ConvBuffer1x1&) = delete;
43 ConvBuffer1x1& operator=(const ConvBuffer1x1&) = delete;
69 ConvBuffer1x1(const OperationDef& definition, const ConvParams& conv_params);
70 friend ConvBuffer1x1 CreateConvBuffer1x1(const GpuInfo& gpu_info,
74 friend ConvBuffer1x1 CreateConvBuffer1x1(const GpuInfo& gpu_info,
78 friend ConvBuffer1x1 CreateConvBuffer1x1Wino4x4To6x6(
[all …]
Dconv_buffer_1x1.cc80 ConvBuffer1x1::ConvParams GetBestParams(const GpuInfo& gpu_info, in GetBestParams()
84 ConvBuffer1x1::ConvParams conv_params; in GetBestParams()
133 ConvBuffer1x1::ConvParams GetBestParams(const GpuInfo& gpu_info, in GetBestParams()
136 ConvBuffer1x1::ConvParams conv_params; in GetBestParams()
148 ConvBuffer1x1::ConvBuffer1x1(const OperationDef& definition, in ConvBuffer1x1() function in tflite::gpu::ConvBuffer1x1
155 ConvBuffer1x1::ConvBuffer1x1(ConvBuffer1x1&& operation) in ConvBuffer1x1() function in tflite::gpu::ConvBuffer1x1
159 ConvBuffer1x1& ConvBuffer1x1::operator=(ConvBuffer1x1&& operation) { in operator =()
167 std::string ConvBuffer1x1::GenerateConvBuffer1x1( in GenerateConvBuffer1x1()
168 const OperationDef& op_def, const ConvBuffer1x1::ConvParams& conv_params, in GenerateConvBuffer1x1()
302 int3 ConvBuffer1x1::GetGridSize() const { in GetGridSize()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconv_buffer_1x1_test.cc59 ConvBuffer1x1 operation = CreateConvBuffer1x1( in TEST_F()
63 absl::make_unique<ConvBuffer1x1>(std::move(operation)), in TEST_F()
70 TEST_F(OpenCLOperationTest, ConvBuffer1x1) { in TEST_F() argument
96 ConvBuffer1x1 operation = CreateConvBuffer1x1( in TEST_F()
100 absl::make_unique<ConvBuffer1x1>(std::move(operation)), in TEST_F()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
Dconvolution_selector.cc93 ConvBuffer1x1 conv = in SelectConvolutionMali()
95 return absl::make_unique<ConvBuffer1x1>(std::move(conv)); in SelectConvolutionMali()
106 ConvBuffer1x1 conv = in SelectConvolutionWinogradMali()
108 return absl::make_unique<ConvBuffer1x1>(std::move(conv)); in SelectConvolutionWinogradMali()
123 ConvBuffer1x1 conv = CreateConvBuffer1x1DynamicWeights( in SelectConvolutionDynamicWeightsMali()
126 return absl::make_unique<ConvBuffer1x1>(std::move(conv)); in SelectConvolutionDynamicWeightsMali()
Dfully_connected_selector.cc70 ConvBuffer1x1 conv = CreateConvBuffer1x1(gpu_info, op_def, attr); in SelectFullyConnectedMali()
71 return absl::make_unique<ConvBuffer1x1>(std::move(conv)); in SelectFullyConnectedMali()