Home
last modified time | relevance | path

Searched refs:Winograd36To4x4 (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dwinograd.h98 class Winograd36To4x4 : public GPUOperation {
100 Winograd36To4x4() = default;
110 Winograd36To4x4(Winograd36To4x4&& kernel) = default;
111 Winograd36To4x4& operator=(Winograd36To4x4&& kernel) = default;
112 Winograd36To4x4(const Winograd36To4x4&) = delete;
113 Winograd36To4x4& operator=(const Winograd36To4x4&) = delete;
116 explicit Winograd36To4x4(const OperationDef& definition) in Winograd36To4x4() function
118 friend Winograd36To4x4 CreateWinograd36To4x4(
123 Winograd36To4x4 CreateWinograd36To4x4(
Dwinograd.cc494 int3 Winograd36To4x4::GetGridSize() const { in GetGridSize()
498 Winograd36To4x4 CreateWinograd36To4x4( in CreateWinograd36To4x4()
501 Winograd36To4x4 desc(definition); in CreateWinograd36To4x4()
Dwinograd_test_util.cc287 Winograd36To4x4 operation = CreateWinograd36To4x4(op_def, biases); in Winograd36To4x4Test()
289 src_tensor, absl::make_unique<Winograd36To4x4>(std::move(operation)), in Winograd36To4x4Test()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dwinograd_test.cc42 TEST_F(OpenCLOperationTest, Winograd36To4x4) { in TEST_F() argument
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dsimple_selectors.cc216 Winograd36To4x4 operation = CreateWinograd36To4x4(op_def, biases); in SelectWinograd36To4x4()
217 return absl::make_unique<Winograd36To4x4>(std::move(operation)); in SelectWinograd36To4x4()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dconv_test.mm285 absl::make_unique<Winograd36To4x4>(std::move(gpu_op3));