Home
last modified time | relevance | path

Searched refs:Conv2dMethod (Results 1 – 3 of 3) sorted by relevance

/external/executorch/backends/vulkan/runtime/graph/ops/impl/
DConvolution.cpp116 enum class Conv2dMethod : uint8_t { enum
127 const Conv2dMethod method, in get_conv2d_shader()
133 case Conv2dMethod::Depthwise: in get_conv2d_shader()
145 case Conv2dMethod::Pointwise: in get_conv2d_shader()
152 case Conv2dMethod::SlidingWindow: in get_conv2d_shader()
155 case Conv2dMethod::Transposed: in get_conv2d_shader()
171 const Conv2dMethod method) { in get_final_sizes()
179 case Conv2dMethod::Depthwise: in get_final_sizes()
181 case Conv2dMethod::Pointwise: in get_final_sizes()
182 case Conv2dMethod::SlidingWindow: in get_final_sizes()
[all …]
/external/pytorch/aten/src/ATen/native/vulkan/ops/
DConvolution.cpp56 static Conv2dMethod determine_method( in determine_method()
290 const Conv2dMethod method, in get_shader()
376 const Conv2dMethod method, in record_op()
449 const Conv2dMethod method, in record_quantized_op()
515 const Conv2dMethod conv_method) { in pack_weights()
1154 const Conv2dMethod method_ = static_cast<Conv2dMethod>( in run_conv2d_context_impl()
DConvolution.h13 enum Conv2dMethod { enum