Searched refs:dim_indices (Results 1 – 4 of 4) sorted by relevance
185 std::vector<int64> PickDims(gtl::ArraySlice<int64> dim_indices) const { in PickDims() argument186 std::vector<int64> res(dim_indices.size()); in PickDims()187 for (size_t i = 0; i < dim_indices.size(); ++i) { in PickDims()188 res[i] = shape_[dim_indices[i]]; in PickDims()
597 dim_indices = math_ops.range(start, dim_value, step)598 dim_indices *= accum_dim_value601 batch_indices += array_ops.reshape(dim_indices, dim_shape)
5170 dim_indices = gen_math_ops._range(start, dim_value, step)5171 dim_indices *= accum_dim_value5174 batch_indices += reshape(dim_indices, dim_shape)
162 const absl::flat_hash_map<char, int>& dim_indices, in GetDimensionIndicesFromLabel() argument167 indices.push_back(dim_indices.at(label)); in GetDimensionIndicesFromLabel()