Searched refs:induction_var_as_vector (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | gather_expander.cc | 164 HloInstruction* induction_var_as_vector = in GatherLoopBody() local
175 MakeDynamicSliceHlo(start_indices, induction_var_as_vector, {1})); in GatherLoopBody()
182 PadVectorWithZeros(induction_var_as_vector, in GatherLoopBody()
216 induction_var_as_vector, /*zeros_to_prepend=*/0, in GatherLoopBody()
|
| D | scatter_expander.cc | 232 HloInstruction* induction_var_as_vector = in ScatterLoopBody() local
242 MakeDynamicSliceHlo(scatter_indices, induction_var_as_vector, {1})); in ScatterLoopBody()
246 PadVectorWithZeros(induction_var_as_vector, in ScatterLoopBody()
266 induction_var_as_vector, /*zeros_to_prepend=*/0, in ScatterLoopBody()
|