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.cc357 int64 min_broadcasted_dimension = output_rank; in SourceIndexOfBroadcast() local
362 min_broadcasted_dimension = in SourceIndexOfBroadcast()
363 std::min(min_broadcasted_dimension, physical_dim); in SourceIndexOfBroadcast()
368 max_broadcasted_dimension - min_broadcasted_dimension == rank - 1; in SourceIndexOfBroadcast()
377 logical_to_physical[dimension_mapping[i]] - min_broadcasted_dimension) { in SourceIndexOfBroadcast()
389 if (min_broadcasted_dimension > 0) { in SourceIndexOfBroadcast()
391 for (int64 i = min_broadcasted_dimension; i <= max_broadcasted_dimension; in SourceIndexOfBroadcast()