Searched refs:update_array_generator (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | dynamic_update_slice_util.cc | 97 bool is_signed, ElementGenerator update_array_generator, in EmitDynamicUpdateSliceInPlaceImpl() argument
147 update_array_generator(update_index)); in EmitDynamicUpdateSliceInPlaceImpl()
177 ElementGenerator update_array_generator = [&](const IrArray::Index& index) { in EmitDynamicUpdateSliceInPlace() local
183 update_shape, start_indices_generator, is_signed, update_array_generator, in EmitDynamicUpdateSliceInPlace()
220 TF_ASSIGN_OR_RETURN(ElementGenerator update_array_generator, in EmitFusedDynamicUpdateSliceInPlaceImpl()
232 update_shape, start_indices_generator, is_signed, update_array_generator, in EmitFusedDynamicUpdateSliceInPlaceImpl()
|