Searched defs:other_batch_dim (Results 1 – 2 of 2) sorted by relevance
19 const Tensor& other, std::optional<int64_t> other_batch_dim, in _binary_pointwise_batch_rule()98 const Tensor& other, std::optional<int64_t> other_batch_dim, in binary_pointwise_inplace_batch_rule()125 const Tensor& other, std::optional<int64_t> other_batch_dim) { in comparison_pointwise_batch_rule()
173 const Tensor& other, std::optional<int64_t> other_batch_dim, in _binary_pointwise_helper()