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 | 78 StatusOr<HloInstruction*> MakeDynamicSliceHlo(
|
D | hlo_creation_utils.cc | 121 StatusOr<HloInstruction*> MakeDynamicSliceHlo( in MakeDynamicSliceHlo() function
|