Searched defs:tensor_bdim (Results 1 – 3 of 3) sorted by relevance
438 const Tensor& tensor, std::optional<int64_t> tensor_bdim, in has_same_shape()461 const Tensor& tensor, std::optional<int64_t> tensor_bdim, in check_same_shape()
155 const Tensor& tensor, std::optional<int64_t> tensor_bdim, in make_dummy()
259 std::optional<int64_t> tensor_bdim, in expect_at_least_rank()