Searched defs:new_bdim (Results 1 – 1 of 1) sorted by relevance
168 int64_t new_bdim = bdim - how_many_dims_of_size_1_before_bdim; in squeeze__batching_rule() local576 …std::optional<int64_t> new_bdim = bdim_size.has_value() ? std::make_optional((int64_t)0) : std::nu… in cat_batching_rule() local