Searched refs:dims_vec (Results 1 – 1 of 1) sorted by relevance
126 const auto& dims_vec = input_array_a.shape().dims(); in UnrollBatchMatMulRecursion() local127 const int current_dim_size = dims_vec[batch_prefix.size()]; in UnrollBatchMatMulRecursion()130 if (batch_prefix.size() + 3 == dims_vec.size()) { in UnrollBatchMatMulRecursion()278 const auto& dims_vec = input_array_a.shape().dims(); in Run() local280 std::accumulate(dims_vec.begin(), dims_vec.end() - 2, 1, in Run()