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 | 42 const Shape& update_shape, const IndexGenerator& start_indices_generator, in EmitDynamicUpdateSliceInPlaceImpl() argument
52 TF_ASSIGN_OR_RETURN(start_multi_index[i], start_indices_generator(i)); in EmitDynamicUpdateSliceInPlaceImpl()
119 IndexGenerator start_indices_generator = [&](int64 index) { in EmitDynamicUpdateSliceInPlace() local
129 update_shape, start_indices_generator, is_signed, update_array_generator, in EmitDynamicUpdateSliceInPlace()
174 IndexGenerator start_indices_generator = [&](int64 index) { in EmitFusedDynamicUpdateSliceInPlaceImpl() local
181 update_shape, start_indices_generator, is_signed, update_array_generator, in EmitFusedDynamicUpdateSliceInPlaceImpl()
|