Searched refs: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
257 float tensor_value = tensor_tensor(d0, d1, d2); in test_block_io_copy_using_reordered_dimensions_do_not_squeeze()
299 TensorMap<Tensor<float, 4, Layout> > tensor_tensor(tensor_data, tensor_dims); in test_block_io_copy_using_reordered_dimensions_squeeze() local
306 float tensor_value = tensor_tensor(d0, d1, d2, d3); in test_block_io_copy_using_reordered_dimensions_squeeze()
|
| /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
333 BINARY_POINTWISE2(op, tensor_tensor);\ in TORCH_LIBRARY_IMPL()
337 #define BINARY_SCALAR_3(op, tensor_tensor, tensor_scalar, scalar_tensor) \ in TORCH_LIBRARY_IMPL() argument
338 BINARY_POINTWISE2(op, tensor_tensor);\ in TORCH_LIBRARY_IMPL()
|