Searched refs:max_broadcasted_dimension (Results 1 – 1 of 1) sorted by relevance
359 int64 max_broadcasted_dimension = -1; in SourceIndexOfBroadcast() local364 max_broadcasted_dimension = in SourceIndexOfBroadcast()365 std::max(max_broadcasted_dimension, physical_dim); in SourceIndexOfBroadcast()368 max_broadcasted_dimension - min_broadcasted_dimension == rank - 1; in SourceIndexOfBroadcast()383 for (int64 i = max_broadcasted_dimension + 1; i < output_rank; ++i) { in SourceIndexOfBroadcast()391 for (int64 i = min_broadcasted_dimension; i <= max_broadcasted_dimension; in SourceIndexOfBroadcast()