Searched refs:start_indices_generator (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | dynamic_update_slice_util.cc | 96 const Shape& update_shape, const IndexGenerator& start_indices_generator, in EmitDynamicUpdateSliceInPlaceImpl() argument
106 TF_ASSIGN_OR_RETURN(start_multi_index[i], start_indices_generator(i)); in EmitDynamicUpdateSliceInPlaceImpl()
173 IndexGenerator start_indices_generator = [&](int64 index) { in EmitDynamicUpdateSliceInPlace() local
183 update_shape, start_indices_generator, is_signed, update_array_generator, in EmitDynamicUpdateSliceInPlace()
223 IndexGenerator start_indices_generator = in EmitFusedDynamicUpdateSliceInPlaceImpl() local
232 update_shape, start_indices_generator, is_signed, update_array_generator, in EmitFusedDynamicUpdateSliceInPlaceImpl()
|