Searched refs:next_output_dim (Results 1 – 2 of 2) sorted by relevance
1325 int64 next_output_dim = 0; in GetShardingFromUser() local1329 user_sharding.tile_assignment().dim(next_output_dim++); in GetShardingFromUser()
45 int64 next_output_dim = 0; in HandleDot() local50 mapping.batch_dims.back().output = next_output_dim++; in HandleDot()66 mapping.lhs_non_contracting_dims.back().output = next_output_dim++; in HandleDot()76 mapping.rhs_non_contracting_dims.back().output = next_output_dim++; in HandleDot()