Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.cc802 absl::Span<const int64> collapsed_or_inserted_dims, in PassthroughOperandToGatherOutputOrScatterUpdate() argument
813 if (absl::c_linear_search(collapsed_or_inserted_dims, i) || in PassthroughOperandToGatherOutputOrScatterUpdate()
847 absl::Span<const int64> collapsed_or_inserted_dims, in PassthroughGatherOutputOrScatterUpdateToOperand() argument
857 if (absl::c_linear_search(collapsed_or_inserted_dims, i) || in PassthroughGatherOutputOrScatterUpdateToOperand()