Searched refs:batch_dim_if_not_empty (Results 1 – 1 of 1) sorted by relevance
421 std::optional<int64_t> batch_dim_if_not_empty(const Tensor& t) { in batch_dim_if_not_empty() function445 const auto res_1_bdim = batch_dim_if_not_empty(res_1); in linalg_lstsq_batch_rule()446 const auto res_2_bdim = batch_dim_if_not_empty(res_2); in linalg_lstsq_batch_rule()447 const auto res_3_bdim = batch_dim_if_not_empty(res_3); in linalg_lstsq_batch_rule()