Searched refs:b_dims_ (Results 1 – 2 of 2) sorted by relevance
185 std::vector<int64_t> b_dims_; variable
1064 b_dims_ = Permute(a_dims_, permutation_); in Initialize()1065 ComputeStrides(elem_size_in_bytes_, b_dims_, b_tiling_, ldb_, ldb_tile_); in Initialize()1092 b_stride1_size = b_dims_.back(); in Initialize()1094 b_stride1_size = std::min(b_stride1_size, b_dims_.back()); in Initialize()