Home
last modified time | relevance | path

Searched refs:fused_emitter (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Ddynamic_update_slice_util.cc193 FusedIrEmitter* fused_emitter, in EmitFusedDynamicUpdateSliceInPlaceImpl() argument
221 fused_emitter->GetGenerator(update)); in EmitFusedDynamicUpdateSliceInPlaceImpl()
227 fused_emitter->GetGenerator(dynamic_update_slice->operand(2 + index))); in EmitFusedDynamicUpdateSliceInPlaceImpl()
238 FusedIrEmitter* fused_emitter, in EmitFusedDynamicUpdateSliceInPlace() argument
241 fusion->called_computations()[0], fusion_output_array, fused_emitter, in EmitFusedDynamicUpdateSliceInPlace()
247 FusedIrEmitter* fused_emitter, in EmitParallelFusedDynamicUpdateSliceInPlace() argument
250 fusion, fusion_output_array, fused_emitter, &launch_dimensions, b); in EmitParallelFusedDynamicUpdateSliceInPlace()
Ddynamic_update_slice_util.h77 FusedIrEmitter* fused_emitter,
84 FusedIrEmitter* fused_emitter,
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc582 FusedIrEmitter fused_emitter(&elemental_emitter); in HandleFusion() local
583 BindFusionArguments(fusion, &fused_emitter); in HandleFusion()
584 TF_ASSIGN_OR_RETURN(auto generator, fused_emitter.GetGenerator( in HandleFusion()
686 FusedIrEmitter* fused_emitter) { in BindFusionArguments() argument
689 fused_emitter->BindGenerator( in BindFusionArguments()
Dir_emitter_unnested.h589 HloComputation* fused_computation, FusedIrEmitter* fused_emitter,
602 HloComputation* fused_computation, FusedIrEmitter* fused_emitter,
621 FusedIrEmitter* fused_emitter,
Dir_emitter_unnested.cc1796 FusedIrEmitter fused_emitter(&elemental_emitter); in EmitLoopFusionFromMlir() local
1801 fused_emitter.BindGenerator( in EmitLoopFusionFromMlir()
1809 fused_emitter.GetGenerator(fused_computation->root_instruction())); in EmitLoopFusionFromMlir()
2048 FusedIrEmitter fused_emitter(&elemental_emitter); in HandleFusion() local
2057 fused_emitter.BindGenerator( in HandleFusion()
2066 fused_computation, output_array, &fused_emitter, launch_dimensions, in HandleFusion()
3731 FusedIrEmitter fused_emitter(&elemental_emitter); in BuildFusedInitializerThunkForMlir() local
3733 fused_emitter.BindGenerator( in BuildFusedInitializerThunkForMlir()
3747 fused_emitter.GetGenerator(instr->operand(1))); in BuildFusedInitializerThunkForMlir()
4155 FusedIrEmitter fused_emitter(&elem_emitter); in EmitTileElementForFusion() local
[all …]
Dir_emitter.h185 FusedIrEmitter* fused_emitter);
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2068 FusedIrEmitter fused_emitter(&elemental_emitter); in HandleFusion() local
2069 BindFusionArguments(fusion, &fused_emitter); in HandleFusion()
2074 fusion, GetIrArrayFor(fusion), &fused_emitter, &b_); in HandleFusion()
2078 FusedIrEmitter fused_emitter(&elemental_emitter); in HandleFusion() local
2079 BindFusionArguments(fusion, &fused_emitter); in HandleFusion()
2080 TF_ASSIGN_OR_RETURN(auto generator, fused_emitter.GetGenerator( in HandleFusion()
3323 FusedIrEmitter* fused_emitter) { in BindFusionArguments() argument
3326 fused_emitter->BindGenerator( in BindFusionArguments()
Dir_emitter.h233 FusedIrEmitter* fused_emitter);