Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dwinograd.h31 class Winograd4x4To36 : public GPUOperation {
33 Winograd4x4To36() = default;
44 Winograd4x4To36(Winograd4x4To36&& kernel) = default;
45 Winograd4x4To36& operator=(Winograd4x4To36&& kernel) = default;
46 Winograd4x4To36(const Winograd4x4To36&) = delete;
47 Winograd4x4To36& operator=(const Winograd4x4To36&) = delete;
50 Winograd4x4To36(const OperationDef& definition, const Padding2D& padding) in Winograd4x4To36() function
52 friend Winograd4x4To36 CreateWinograd4x4To36(const OperationDef& definition,
58 Winograd4x4To36 CreateWinograd4x4To36(const OperationDef& definition,
Dwinograd.cc189 int3 Winograd4x4To36::GetGridSize() const { in GetGridSize()
199 absl::Status Winograd4x4To36::BindArguments(ArgumentsBinder* args) { in BindArguments()
211 Winograd4x4To36 CreateWinograd4x4To36(const OperationDef& definition, in CreateWinograd4x4To36()
213 Winograd4x4To36 desc(definition, padding); in CreateWinograd4x4To36()
Dwinograd_test_util.cc219 Winograd4x4To36 operation = CreateWinograd4x4To36(op_def, padding); in Winograd4x4To36Test()
221 src_tensor, absl::make_unique<Winograd4x4To36>(std::move(operation)), in Winograd4x4To36Test()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dwinograd_test.cc37 TEST_F(OpenCLOperationTest, Winograd4x4To36) { in TEST_F() argument
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dsimple_selectors.cc198 Winograd4x4To36 operation = CreateWinograd4x4To36(op_def, padding); in SelectWinograd4x4To36()
199 return absl::make_unique<Winograd4x4To36>(std::move(operation)); in SelectWinograd4x4To36()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dconv_test.mm277 absl::make_unique<Winograd4x4To36>(std::move(gpu_op1));