Searched defs:tensor_scalar (Results 1 – 1 of 1) sorted by relevance
332 #define BINARY_SCALAR_2(op, tensor_tensor, tensor_scalar) \ in TORCH_LIBRARY_IMPL() argument337 #define BINARY_SCALAR_3(op, tensor_tensor, tensor_scalar, scalar_tensor) \ in TORCH_LIBRARY_IMPL() argument342 #define BINARY_SCALAR_3_Tensor(op, tensor_scalar, scalar_tensor) \ in TORCH_LIBRARY_IMPL() argument