Searched refs:PadVectorWithZeros (Results 1 – 5 of 5) sorted by relevance
166 TEST_F(HloCreationUtilsTest, PadVectorWithZeros) { in TEST_F() argument176 PadVectorWithZeros(param, /*zeros_to_prepend=*/3, /*zeros_to_append=*/1)); in TEST_F()
182 PadVectorWithZeros(induction_var_as_vector, in GatherLoopBody()215 PadVectorWithZeros( in GatherLoopBody()
246 PadVectorWithZeros(induction_var_as_vector, in ScatterLoopBody()265 PadVectorWithZeros( in ScatterLoopBody()
209 StatusOr<HloInstruction*> PadVectorWithZeros(HloInstruction* operand,
420 StatusOr<HloInstruction*> PadVectorWithZeros(HloInstruction* operand, in PadVectorWithZeros() function