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 | 192 const HloComputation* fusion, const IrArray& fusion_output_array, in EmitFusedDynamicUpdateSliceInPlaceImpl() argument
233 fusion_output_array, launch_dimensions, IrName(dynamic_update_slice), b); in EmitFusedDynamicUpdateSliceInPlaceImpl()
237 const IrArray& fusion_output_array, in EmitFusedDynamicUpdateSliceInPlace() argument
241 fusion->called_computations()[0], fusion_output_array, fused_emitter, in EmitFusedDynamicUpdateSliceInPlace()
246 const HloComputation* fusion, const IrArray& fusion_output_array, in EmitParallelFusedDynamicUpdateSliceInPlace() argument
250 fusion, fusion_output_array, fused_emitter, &launch_dimensions, b); in EmitParallelFusedDynamicUpdateSliceInPlace()
|
| D | dynamic_update_slice_util.h | 76 const IrArray& fusion_output_array,
83 const HloComputation* fusion, const IrArray& fusion_output_array,
|