Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.h227 Status HandleAfterAll(HloInstruction* after_all) override;
Dhlo_evaluator.cc1165 Status HloEvaluator::HandleAfterAll(HloInstruction* after_all) { in HandleAfterAll() argument
1166 evaluated_[after_all] = LiteralUtil::CreateToken(); in HandleAfterAll()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h182 Status HandleAfterAll(HloInstruction* after_all) override;
Dir_emitter_unnested.cc1469 Status IrEmitterUnnested::HandleAfterAll(HloInstruction* after_all) { in HandleAfterAll() argument
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h182 Status HandleAfterAll(HloInstruction* after_all) override;
Dir_emitter.cc2638 Status IrEmitter::HandleAfterAll(HloInstruction* after_all) { in HandleAfterAll() argument
2639 TF_RET_CHECK(ByteSizeOf(after_all->shape()) == 0); in HandleAfterAll()
2641 TF_RETURN_IF_ERROR(EmitTargetAddressForOp(after_all)); in HandleAfterAll()