Home
last modified time | relevance | path

Searched refs:MakeDynamicSliceHlo (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dscatter_expander.cc242 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()
Dgather_expander.cc175 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()
Dhlo_creation_utils.h78 StatusOr<HloInstruction*> MakeDynamicSliceHlo(
Dhlo_creation_utils.cc121 StatusOr<HloInstruction*> MakeDynamicSliceHlo( in MakeDynamicSliceHlo() function