Searched refs:passthrough_dims (Results 1 – 1 of 1) sorted by relevance
423 absl::Span<const ReshapePassthroughDimPair> passthrough_dims, int64 dim) { in IsReshapePassthroughOperandDim() argument424 return absl::c_any_of(passthrough_dims, in IsReshapePassthroughOperandDim()433 absl::Span<const ReshapePassthroughDimPair> passthrough_dims, in MapPassthroughOperandDimToResultDim() argument436 passthrough_dims, [&](ReshapePassthroughDimPair passthrough_dim_pair) { in MapPassthroughOperandDimToResultDim()439 CHECK(it != passthrough_dims.end()); in MapPassthroughOperandDimToResultDim()