Searched refs:max_broadcasted_dimension (Results 1 – 1 of 1) sorted by relevance
272 int64 max_broadcasted_dimension = -1; in SourceIndexOfBroadcast() local277 max_broadcasted_dimension = in SourceIndexOfBroadcast()278 std::max(max_broadcasted_dimension, physical_dim); in SourceIndexOfBroadcast()281 max_broadcasted_dimension - min_broadcasted_dimension == rank - 1; in SourceIndexOfBroadcast()296 for (int64 i = max_broadcasted_dimension + 1; i < output_rank; ++i) { in SourceIndexOfBroadcast()304 for (int64 i = min_broadcasted_dimension; i <= max_broadcasted_dimension; in SourceIndexOfBroadcast()