Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dlogical_buffer_analysis.cc185 Status LogicalBufferAnalysis::HandleTupleSelect(HloInstruction* tuple_select) { in HandleTupleSelect() argument
188 NewLogicalBuffer(tuple_select, /*index=*/{}); in HandleTupleSelect()
Dtuple_points_to_analysis.cc445 Status TuplePointsToAnalysis::HandleTupleSelect(HloInstruction* tuple_select) { in HandleTupleSelect() argument
453 auto on_true = tuple_select->operand(1); in HandleTupleSelect()
454 auto on_false = tuple_select->operand(2); in HandleTupleSelect()
455 PointsToSet& points_to_set = CreateCopiedPointsToSet(tuple_select, on_true); in HandleTupleSelect()
473 logical_buffer_analysis_->GetBuffer(tuple_select, /*index=*/{}), in HandleTupleSelect()
Dlogical_buffer_analysis.h69 Status HandleTupleSelect(HloInstruction* tuple_select) override;
Ddfs_hlo_visitor_with_default.h83 Status HandleTupleSelect(HloInstructionPtr tuple_select) override { in HandleTupleSelect() argument
84 return DefaultAction(tuple_select); in HandleTupleSelect()
Dtuple_points_to_analysis.h257 Status HandleTupleSelect(HloInstruction* tuple_select) override;
Dhlo_evaluator.h245 Status HandleTupleSelect(HloInstruction* tuple_select) override;
Dhlo_verifier.h48 Status HandleTupleSelect(HloInstruction* tuple_select) override;
Dhlo_evaluator.cc2018 Status HloEvaluator::HandleTupleSelect(HloInstruction* tuple_select) { in HandleTupleSelect() argument
2019 const auto& pred = GetEvaluatedLiteralFor(tuple_select->operand(0)); in HandleTupleSelect()
2020 const auto& on_true = GetEvaluatedLiteralFor(tuple_select->operand(1)); in HandleTupleSelect()
2021 const auto& on_false = GetEvaluatedLiteralFor(tuple_select->operand(2)); in HandleTupleSelect()
2024 evaluated_[tuple_select] = on_true.Clone(); in HandleTupleSelect()
2026 evaluated_[tuple_select] = on_false.Clone(); in HandleTupleSelect()
Dhlo_verifier.cc110 Status ShapeVerifier::HandleTupleSelect(HloInstruction* tuple_select) { in HandleTupleSelect() argument
111 return CheckTernaryShape(tuple_select); in HandleTupleSelect()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.h93 Status HandleTupleSelect(HloInstruction* tuple_select) override;
Dir_emitter.cc517 Status IrEmitter::HandleTupleSelect(HloInstruction* tuple_select) { in HandleTupleSelect() argument
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc375 Status IrEmitter::HandleTupleSelect(HloInstruction* tuple_select) { in HandleTupleSelect() argument
376 auto pred = tuple_select->operand(0); in HandleTupleSelect()
377 auto on_true = tuple_select->operand(1); in HandleTupleSelect()
378 auto on_false = tuple_select->operand(2); in HandleTupleSelect()
381 TF_RET_CHECK(tuple_select->shape().IsTuple()); in HandleTupleSelect()
382 TF_RETURN_IF_ERROR(EmitTargetAddressForOp(tuple_select)); in HandleTupleSelect()
383 llvm_ir::EmitTupleSelect(GetIrArrayFor(tuple_select), GetIrArrayFor(pred), in HandleTupleSelect()
Dir_emitter.h138 Status HandleTupleSelect(HloInstruction* tuple_select) override;