Searched refs:MakeDynamicSliceHlo (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | scatter_expander.cc | 242 MakeDynamicSliceHlo(scatter_indices, induction_var_as_vector, {1})); in ScatterLoopBody()
251 MakeDynamicSliceHlo(scatter_indices, index_into_scatter_indices, in ScatterLoopBody()
273 MakeDynamicSliceHlo(updates, index_into_updates, update_slice_bounds)); in ScatterLoopBody()
294 MakeDynamicSliceHlo(operand, scatter_slice_start, in ScatterLoopBody()
|
| D | gather_expander.cc | 175 MakeDynamicSliceHlo(start_indices, induction_var_as_vector, {1})); in GatherLoopBody()
188 MakeDynamicSliceHlo(start_indices, index_into_start_indices, in GatherLoopBody()
201 MakeDynamicSliceHlo(operand, gathered_slice_start, in GatherLoopBody()
|
| D | hlo_creation_utils.h | 87 StatusOr<HloInstruction*> MakeDynamicSliceHlo(
90 StatusOr<HloInstruction*> MakeDynamicSliceHlo(
|
| D | hlo_creation_utils.cc | 135 StatusOr<HloInstruction*> MakeDynamicSliceHlo( in MakeDynamicSliceHlo() function
150 StatusOr<HloInstruction*> MakeDynamicSliceHlo( in MakeDynamicSliceHlo() function
|