Searched refs:max_logical_bdims (Results 1 – 1 of 1) sorted by relevance
467 …const auto max_logical_bdims = std::max({input_logical_num_bdims, atol_logical_num_bdims, rtol_log… in atol_rtol_tensor_batch_rule() local474 input_ = maybePadToLogicalRank(input_, input_bdim, max_logical_bdims + 2); in atol_rtol_tensor_batch_rule()475 atol_ = atol_.has_value() ? maybePadToLogicalRank(*atol_, atol_bdim, max_logical_bdims) : atol_; in atol_rtol_tensor_batch_rule()476 rtol_ = rtol_.has_value() ? maybePadToLogicalRank(*rtol_, rtol_bdim, max_logical_bdims) : rtol_; in atol_rtol_tensor_batch_rule()