Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dindexed_array_analysis.cc423 absl::Span<const ReshapePassthroughDimPair> passthrough_dims, int64 dim) { in IsReshapePassthroughOperandDim() argument
424 return absl::c_any_of(passthrough_dims, in IsReshapePassthroughOperandDim()
433 absl::Span<const ReshapePassthroughDimPair> passthrough_dims, in MapPassthroughOperandDimToResultDim() argument
436 passthrough_dims, [&](ReshapePassthroughDimPair passthrough_dim_pair) { in MapPassthroughOperandDimToResultDim()
439 CHECK(it != passthrough_dims.end()); in MapPassthroughOperandDimToResultDim()