Searched refs:ReshapePassthroughDimPair (Results 1 – 1 of 1) sorted by relevance
332 struct ReshapePassthroughDimPair { struct344 std::vector<ReshapePassthroughDimPair> ComputeReshapePassthroughDimPairs( in ComputeReshapePassthroughDimPairs()368 std::vector<ReshapePassthroughDimPair> result; in ComputeReshapePassthroughDimPairs()397 [](ReshapePassthroughDimPair value) { in ComputeReshapePassthroughDimPairs()408 result, [](ReshapePassthroughDimPair lhs, ReshapePassthroughDimPair rhs) { in ComputeReshapePassthroughDimPairs()413 result, [](ReshapePassthroughDimPair lhs, ReshapePassthroughDimPair rhs) { in ComputeReshapePassthroughDimPairs()423 absl::Span<const ReshapePassthroughDimPair> passthrough_dims, int64 dim) { in IsReshapePassthroughOperandDim()425 [&](ReshapePassthroughDimPair passthrough_dim_pair) { in IsReshapePassthroughOperandDim()433 absl::Span<const ReshapePassthroughDimPair> passthrough_dims, in MapPassthroughOperandDimToResultDim()436 passthrough_dims, [&](ReshapePassthroughDimPair passthrough_dim_pair) { in MapPassthroughOperandDimToResultDim()[all …]