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 | 43 bool is_signed, ElementGenerator update_array_generator, in EmitDynamicUpdateSliceInPlaceImpl() argument
93 update_array_generator(update_index)); in EmitDynamicUpdateSliceInPlaceImpl()
123 ElementGenerator update_array_generator = [&](const IrArray::Index& index) { in EmitDynamicUpdateSliceInPlace() local
129 update_shape, start_indices_generator, is_signed, update_array_generator, in EmitDynamicUpdateSliceInPlace()
172 ElementGenerator update_array_generator = fused_emitter.GetGenerator(update); in EmitFusedDynamicUpdateSliceInPlaceImpl() local
181 update_shape, start_indices_generator, is_signed, update_array_generator, in EmitFusedDynamicUpdateSliceInPlaceImpl()
|