Searched refs:resize_out_tensor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/executorch/kernels/quantized/cpu/ |
| H A D | op_embedding.cpp | 182 void resize_out_tensor( in resize_out_tensor() function
264 resize_out_tensor(weight, indices, out); in quantized_embedding_byte_out()
328 resize_out_tensor(weight, indices, out); in quantized_embedding_byte_dtype_out()
|
| H A D | embeddingxb.cpp | 223 void resize_out_tensor( in resize_out_tensor() function
311 resize_out_tensor(weight, indices, out, weight_nbit); in quantized_embedding_xbit_out()
381 resize_out_tensor(weight, indices, out, weight_nbit); in quantized_embedding_xbit_dtype_out()
|
| /aosp_15_r20/external/executorch/kernels/optimized/cpu/ |
| H A D | op_bmm.cpp | 113 Error resize_out_tensor(const Tensor& self, const Tensor& mat2, Tensor& out) { in resize_out_tensor() function
148 resize_out_tensor(self, mat2, out) == Error::Ok, in opt_bmm_out()
|