Searched defs:tensor_batch_dim (Results 1 – 2 of 2) sorted by relevance
18 const Tensor& tensor, std::optional<int64_t> tensor_batch_dim, in _binary_pointwise_batch_rule()97 Tensor& tensor, std::optional<int64_t> tensor_batch_dim, in binary_pointwise_inplace_batch_rule()124 const Tensor& tensor, std::optional<int64_t> tensor_batch_dim, in comparison_pointwise_batch_rule()
172 const Tensor& tensor, std::optional<int64_t> tensor_batch_dim, in _binary_pointwise_helper()