Searched refs:tensor_scalar (Results 1 – 1 of 1) sorted by relevance
| /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
334 UNARY_POINTWISE2(op, tensor_scalar); in TORCH_LIBRARY_IMPL()
337 #define BINARY_SCALAR_3(op, tensor_tensor, tensor_scalar, scalar_tensor) \ in TORCH_LIBRARY_IMPL() argument
339 UNARY_POINTWISE2(op, tensor_scalar);\ in TORCH_LIBRARY_IMPL()
342 #define BINARY_SCALAR_3_Tensor(op, tensor_scalar, scalar_tensor) \ in TORCH_LIBRARY_IMPL() argument
344 UNARY_POINTWISE2(op, tensor_scalar);\ in TORCH_LIBRARY_IMPL()
|