Home
last modified time | relevance | path

Searched defs:mat2_data (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DMatMul.cpp24 const ValueRef mat2_data, in check_matmul_args()
62 const ValueRef mat2_data, in add_matmul_naive_buffer_node()
113 const ValueRef mat2_data, in add_matmul_naive_texture3d_node()
158 const ValueRef mat2_data, in add_matmul_optimized_node()
244 const ValueRef mat2_data, in add_matmul_node()
H A DLinear.cpp25 const ValueRef mat2_data, in check_addmm_args()
91 const ValueRef mat2_data, in add_addmm_naive_node()
141 const ValueRef mat2_data, in add_addmm_optimized_node()
H A DQuantizedLinear.cpp236 const ValueRef mat2_data, in check_q_4w_linear_args()
295 const ValueRef mat2_data, in add_q_4w_linear_node()
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_bmm.cpp43 const CTYPE* mat2_data = mat2.const_data_ptr<CTYPE>(); in bmm_out() local
H A Dvec_ops.h151 const T* __restrict__ mat2_data, in vec_addmm()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DSampledAddmmKernel.cpp30 const scalar_t* mat2_data = mat2.const_data_ptr<scalar_t>(); in sampled_addmm_sparse_csr_kernel_impl() local
/aosp_15_r20/external/executorch/backends/vulkan/test/
H A Dvulkan_compute_api_test.cpp780 std::vector<float> mat2_data = in TEST_F() local