Searched refs:from_indices (Results 1 – 2 of 2) sorted by relevance
405 ConvDimIndices from_indices = GetDimIndices(from, input.size()); in ReorderDims() local409 reordered[to_indices.data.batch_idx] = input[from_indices.data.batch_idx]; in ReorderDims()410 reordered[to_indices.data.depth_idx] = input[from_indices.data.depth_idx]; in ReorderDims()412 int spatial_idx_from = from_indices.data.spatial_idx; in ReorderDims()427 ConvDimIndices from_indices = GetDimIndices(from, input.size()); in ReorderDims() local432 input[from_indices.filter.output_idx]; in ReorderDims()433 reordered[to_indices.filter.input_idx] = input[from_indices.filter.input_idx]; in ReorderDims()435 int spatial_idx_from = from_indices.filter.spatial_idx; in ReorderDims()
1586 auto from_indices = hlo_sharding_util::ScatterDataSharding( in GetShardingFromUser() local1588 if (!from_indices.IsTileMaximal()) { in GetShardingFromUser()1589 return from_indices; in GetShardingFromUser()