Searched defs:tensor_tensor (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_block_io.cpp | 251 TensorMap<Tensor<float, 3, Layout> > tensor_tensor(tensor_data, tensor_dims); in test_block_io_copy_using_reordered_dimensions_do_not_squeeze() local
299 TensorMap<Tensor<float, 4, Layout> > tensor_tensor(tensor_data, tensor_dims); in test_block_io_copy_using_reordered_dimensions_squeeze() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesBinaryOps.cpp | 332 #define BINARY_SCALAR_2(op, tensor_tensor, tensor_scalar) \ in TORCH_LIBRARY_IMPL() argument
337 #define BINARY_SCALAR_3(op, tensor_tensor, tensor_scalar, scalar_tensor) \ in TORCH_LIBRARY_IMPL() argument
|