| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | Convolution.cpp | 39 const auto& weight_sizes = weight_ref->sizes; in resize_conv2d_node() local
69 const std::vector<int64_t>& weight_sizes = weight_ref->sizes; in resize_conv1d_node() local
136 const auto& weight_sizes = graph.get_tref(weight)->sizes; in get_conv2d_shader() local
248 const auto weight_sizes = graph.sizes_of(weight); in create_conv2d_params() local
273 const auto weight_sizes = graph.sizes_of(weight); in get_conv2d_method() local
435 std::vector<int64_t> weight_sizes = t_weight->sizes(); in add_conv1d_node() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/ |
| H A D | conv.h | 77 std::vector<int64_t> weight_sizes = { in reset() local
85 std::vector<int64_t> weight_sizes = { in reset() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qembeddingbag.cpp | 52 const auto weight_sizes = weight.sizes(); in embedding_lookup_fallback_impl() local
256 const auto weight_sizes = weight.sizes(); in embedding_bag_nbit_impl() local
428 const auto weight_sizes = weight.sizes(); in embedding_bag_byte_impl() local
|
| H A D | qembeddingbag_prepack.cpp | 241 const auto weight_sizes = weight.sizes(); in qembeddingbag_byte_prepack_out() local
334 const auto weight_sizes = weight.sizes(); in qembeddingbag_byte_prepack_meta() local
|
| H A D | QnnpackUtils.h | 64 std::vector<int64_t> weight_sizes; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cuda/ |
| H A D | EmbeddingBag.cu | 217 const auto weight_sizes = weight.sizes(); in embedding_bag_byte_impl() local
399 const auto weight_sizes = weight.sizes(); in embedding_bag_4bit_impl() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Convolution.cpp | 657 const c10::ArrayRef<T>& weight_sizes, const at::Tensor& bias, in check_shape_forward()
740 const c10::ArrayRef<T>& weight_sizes, in check_shape_backward()
999 auto weight_sizes = weight.sym_sizes(); in convolution_same() local
1481 c10::IntArrayRef weight_sizes = weight.sizes(); in _convolution() local
|
| H A D | Histogram.cpp | 112 auto weight_sizes = weight.value().sizes().vec(); in histogramdd_check_inputs() local
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/utils/ |
| H A D | KernelUtils.cpp | 24 const auto weight_sizes = graph.get_tref(weight)->sizes; in make_ivec2_kernel_size() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/xnnpack/ |
| H A D | Convolution.cpp | 208 std::array<int64_t, 4> weight_sizes; in create() local
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_convolution.cpp | 204 SizesArrayRef weight_sizes = weight.sizes(); in convolution_wrapper() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | ConvolutionMM2d.cu | 230 auto weight_sizes = weight.sizes(); in slow_conv2d_backward() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/ |
| H A D | Conv.cpp | 147 std::vector<int64_t> weight_sizes(weight_dim); in check_shape_forward() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Conv.cpp | 64 const IntArrayRef& weight_sizes = weight.sizes(); in check_shape_forward() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Convolution.cpp | 834 const IntArrayRef& weight_sizes = weight.sizes(); in run_conv1d_context_impl() local
|