Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils_test.cc166 TEST_F(HloCreationUtilsTest, PadVectorWithZeros) { in TEST_F() argument
176 PadVectorWithZeros(param, /*zeros_to_prepend=*/3, /*zeros_to_append=*/1)); in TEST_F()
Dgather_expander.cc182 PadVectorWithZeros(induction_var_as_vector, in GatherLoopBody()
215 PadVectorWithZeros( in GatherLoopBody()
Dscatter_expander.cc246 PadVectorWithZeros(induction_var_as_vector, in ScatterLoopBody()
265 PadVectorWithZeros( in ScatterLoopBody()
Dhlo_creation_utils.h209 StatusOr<HloInstruction*> PadVectorWithZeros(HloInstruction* operand,
Dhlo_creation_utils.cc420 StatusOr<HloInstruction*> PadVectorWithZeros(HloInstruction* operand, in PadVectorWithZeros() function