Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Ddepthwise_conv_3x3.h36 class DepthwiseConv3x3 : public GPUOperation {
38 DepthwiseConv3x3() = default;
46 DepthwiseConv3x3(DepthwiseConv3x3&& operation);
47 DepthwiseConv3x3& operator=(DepthwiseConv3x3&& operation);
48 DepthwiseConv3x3(const DepthwiseConv3x3&) = delete;
49 DepthwiseConv3x3& operator=(const DepthwiseConv3x3&) = delete;
52 explicit DepthwiseConv3x3(const OperationDef& definition,
60 friend DepthwiseConv3x3 CreateDepthwiseConv3x3(
77 void DepthwiseConv3x3::UploadWeightsAndBiases( in UploadWeightsAndBiases()
117 void DepthwiseConv3x3::RearrangeWeightsAndBiasesData( in RearrangeWeightsAndBiasesData()
[all …]
Ddepthwise_conv_3x3.cc27 DepthwiseConv3x3::DepthwiseConv3x3(const OperationDef& definition, in DepthwiseConv3x3() function in tflite::gpu::DepthwiseConv3x3
42 DepthwiseConv3x3::DepthwiseConv3x3(DepthwiseConv3x3&& operation) in DepthwiseConv3x3() function in tflite::gpu::DepthwiseConv3x3
46 DepthwiseConv3x3& DepthwiseConv3x3::operator=(DepthwiseConv3x3&& operation) { in operator =()
54 std::string DepthwiseConv3x3::GenerateDepthwiseConvCode( in GenerateDepthwiseConvCode()
284 int3 DepthwiseConv3x3::GetGridSize() const { in GetGridSize()
291 void DepthwiseConv3x3::GetPossibleKernelWorkGroups( in GetPossibleKernelWorkGroups()
311 DepthwiseConv3x3 CreateDepthwiseConv3x3( in CreateDepthwiseConv3x3()
318 DepthwiseConv3x3 result(definition, weights_are_buffer, local_mem_uploads, in CreateDepthwiseConv3x3()
Ddepthwise_conv_3x3_test_util.cc53 DepthwiseConv3x3 operation = in DepthwiseConv3x3SimpleWeightsTest()
56 src_tensor, absl::make_unique<DepthwiseConv3x3>(std::move(operation)), in DepthwiseConv3x3SimpleWeightsTest()
91 DepthwiseConv3x3 operation = in DepthwiseConv3x3Test()
94 src_tensor, absl::make_unique<DepthwiseConv3x3>(std::move(operation)), in DepthwiseConv3x3Test()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
Ddw_convolution_selector.cc30 return absl::make_unique<DepthwiseConv3x3>( in SelectDWConvolutionAdreno()
42 return absl::make_unique<DepthwiseConv3x3>( in SelectDWConvolutionPowerVR()
59 return absl::make_unique<DepthwiseConv3x3>( in SelectDWConvolutionMali()
71 return absl::make_unique<DepthwiseConv3x3>( in SelectDWConvolutionApple()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Ddepthwise_conv_3x3_test.cc36 TEST_F(OpenCLOperationTest, DepthwiseConv3x3) { in TEST_F() argument