Searched refs:fusion_output_array (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | dynamic_update_slice_util.cc | 140 const IrArray& fusion_output_array, ElementalIrEmitter* elemental_emitter, in EmitFusedDynamicUpdateSliceInPlaceImpl() argument
182 fusion_output_array, launch_dimensions, IrName(fusion), b); in EmitFusedDynamicUpdateSliceInPlaceImpl()
188 const IrArray& fusion_output_array, ElementalIrEmitter* elemental_emitter, in EmitFusedDynamicUpdateSliceInPlace() argument
191 fusion, std::move(operand_arrays_generator), fusion_output_array, in EmitFusedDynamicUpdateSliceInPlace()
199 const IrArray& fusion_output_array, ElementalIrEmitter* elemental_emitter, in EmitParallelFusedDynamicUpdateSliceInPlace() argument
202 fusion, std::move(operand_arrays_generator), fusion_output_array, in EmitParallelFusedDynamicUpdateSliceInPlace()
|
| D | dynamic_update_slice_util.h | 81 const IrArray& fusion_output_array, ElementalIrEmitter* elemental_emitter,
89 const IrArray& fusion_output_array, ElementalIrEmitter* elemental_emitter,
|