Searched refs:b_tiling_ (Results 1 – 2 of 2) sorted by relevance
662 size *= RoundUpTo(a_dims_[permutation_[i]], b_tiling_[i]); in OutputNumElems()757 node.ldb = b_tiling_[pos_stride1a_in_b] > 1 in BuildPlanNodes()769 DCHECK(a_tiling_[a_dim] == 1 || b_tiling_[b_dim] == 1 || in BuildPlanNodes()770 a_tiling_[a_dim] == b_tiling_[b_dim]); in BuildPlanNodes()771 int64_t tile_size = std::max(a_tiling_[a_dim], b_tiling_[b_dim]); in BuildPlanNodes()785 node.ldb = b_tiling_[b_dim] > 1 ? ldb_tile_[b_dim] : ldb_[b_dim]; in BuildPlanNodes()838 node.ldb = ldb_[b_dim] * tile_size / b_tiling_[b_dim]; in BuildPlanNodes()927 ParseTilingSpecification(ndim, output_tiling.tiling, plan->b_tiling_)); in Create()995 (absl::c_find_if(plan->b_tiling_, is_not_one) != plan->b_tiling_.end()); in Create()1001 absl::StrJoin(plan->b_tiling_, ",")); in Create()[all …]
200 absl::InlinedVector<int64_t, 4> b_tiling_; variable