Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc1911 HloInstruction* init_value_operand = [&] { in BuildInitializerThunk() local
1931 const HloInstruction* init_value = init_value_operand; in BuildInitializerThunk()
2005 TF_RETURN_IF_ERROR(init_value_operand->Accept(&fused_emitter)); in BuildInitializerThunk()
2007 ParallelLoopEmitter(fused_emitter.GetGenerator(init_value_operand), in BuildInitializerThunk()
2676 HloInstruction* init_value_operand = reduce_inst->mutable_operand(1); in EmitPrologueForOneReduction() local
2680 TF_CHECK_OK(init_value_operand->Accept(&fused_emitter)); in EmitPrologueForOneReduction()
2683 .GetGenerator(init_value_operand)(IrArray::Index(b_.getInt32Ty())) in EmitPrologueForOneReduction()