Home
last modified time | relevance | path

Searched refs:min_broadcasted_dimension (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.cc270 int64 min_broadcasted_dimension = output_rank; in SourceIndexOfBroadcast() local
275 min_broadcasted_dimension = in SourceIndexOfBroadcast()
276 std::min(min_broadcasted_dimension, physical_dim); in SourceIndexOfBroadcast()
281 max_broadcasted_dimension - min_broadcasted_dimension == rank - 1; in SourceIndexOfBroadcast()
290 logical_to_physical[dimension_mapping[i]] - min_broadcasted_dimension) { in SourceIndexOfBroadcast()
302 if (min_broadcasted_dimension > 0) { in SourceIndexOfBroadcast()
304 for (int64 i = min_broadcasted_dimension; i <= max_broadcasted_dimension; in SourceIndexOfBroadcast()