Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils.cc389 HloInstruction* operand, absl::Span<const int64> dims_to_insert) { in InsertDegenerateDims() argument
390 CHECK(absl::c_is_sorted(dims_to_insert)); in InsertDegenerateDims()
394 operand_shape.dimensions_size() + dims_to_insert.size(); in InsertDegenerateDims()
395 for (auto dim_to_insert : dims_to_insert) { in InsertDegenerateDims()
404 if (dims_to_insert_idx < dims_to_insert.size() && in InsertDegenerateDims()
405 i == dims_to_insert[dims_to_insert_idx]) { in InsertDegenerateDims()
Dhlo_creation_utils.h205 HloInstruction* operand, absl::Span<const int64> dims_to_insert);