Searched refs:PrependDegenerateDims (Results 1 – 5 of 5) sorted by relevance
97 PrependDegenerateDims(param, 1)); in TEST_F()116 PrependDegenerateDims(param, 2)); in TEST_F()135 PrependDegenerateDims(param, 2)); in TEST_F()
74 return PrependDegenerateDims(transposed_start_indices, 1); in CanonicalizeGatherIndices()211 PrependDegenerateDims(gathered_slice_with_dims_collapsed, 1)); in GatherLoopBody()
81 return PrependDegenerateDims(transposed_scatter_indices, 1); in CanonicalizeScatterIndices()125 return PrependDegenerateDims(updates, 1); in AdjustScatterDims()
174 StatusOr<HloInstruction*> PrependDegenerateDims(HloInstruction* operand,
335 StatusOr<HloInstruction*> PrependDegenerateDims(HloInstruction* operand, in PrependDegenerateDims() function