Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Ddynamic_update_slice_util.cc169 FusedIrEmitter fused_emitter(std::move(operand_arrays_generator), in EmitFusedDynamicUpdateSliceInPlaceImpl() local
171 TF_RETURN_IF_ERROR(dynamic_update_slice->Accept(&fused_emitter)); in EmitFusedDynamicUpdateSliceInPlaceImpl()
172 ElementGenerator update_array_generator = fused_emitter.GetGenerator(update); in EmitFusedDynamicUpdateSliceInPlaceImpl()
176 fused_emitter.GetGenerator(dynamic_update_slice->operand(2 + index)); in EmitFusedDynamicUpdateSliceInPlaceImpl()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2003 FusedIrEmitter fused_emitter(GetGeneratorForOperandIrArrays(hlo), in BuildInitializerThunk() local
2005 TF_RETURN_IF_ERROR(init_value_operand->Accept(&fused_emitter)); in BuildInitializerThunk()
2007 ParallelLoopEmitter(fused_emitter.GetGenerator(init_value_operand), in BuildInitializerThunk()
2494 FusedIrEmitter fused_emitter(GetGeneratorForOperandIrArrays(hlo), in EmitTileElementForFusion() local
2498 fused_emitter.SetTiledParameterInfo(tiled_param_info); in EmitTileElementForFusion()
2499 TF_CHECK_OK(hlo->fused_expression_root()->Accept(&fused_emitter)); in EmitTileElementForFusion()
2504 fused_emitter.GetRootGenerator(); in EmitTileElementForFusion()
2677 FusedIrEmitter fused_emitter(GetGeneratorForOperandIrArrays(unnested_hlo), in EmitPrologueForOneReduction() local
2680 TF_CHECK_OK(init_value_operand->Accept(&fused_emitter)); in EmitPrologueForOneReduction()
2682 fused_emitter in EmitPrologueForOneReduction()
[all …]
Dir_emitter.cc728 FusedIrEmitter fused_emitter(GetGeneratorForOperandIrArrays(fusion), in HandleFusion() local
730 TF_RETURN_IF_ERROR(fusion->fused_expression_root()->Accept(&fused_emitter)); in HandleFusion()
732 return EmitTargetElementLoop(*fusion, fused_emitter.GetRootGenerator()); in HandleFusion()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2163 FusedIrEmitter fused_emitter(GetGeneratorForOperandIrArrays(fusion), in HandleFusion() local
2165 TF_RETURN_IF_ERROR(fusion->fused_expression_root()->Accept(&fused_emitter)); in HandleFusion()
2167 return EmitTargetElementLoop(fusion, fused_emitter.GetRootGenerator()); in HandleFusion()