Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc427 Status IrEmitter::HandleTupleSelect(HloInstruction* tuple_select) { in HandleTupleSelect() argument
428 auto pred = tuple_select->operand(0); in HandleTupleSelect()
429 auto on_true = tuple_select->operand(1); in HandleTupleSelect()
430 auto on_false = tuple_select->operand(2); in HandleTupleSelect()
433 TF_RET_CHECK(tuple_select->shape().IsTuple()); in HandleTupleSelect()
434 llvm_ir::EmitTupleSelect(GetIrArray(*tuple_select, *tuple_select), in HandleTupleSelect()
435 GetIrArray(*pred, *tuple_select), in HandleTupleSelect()
Dir_emitter.h96 Status HandleTupleSelect(HloInstruction* tuple_select) override;
Dir_emitter_unnested.h180 Status HandleTupleSelect(HloInstruction* tuple_select) override;
Dir_emitter_unnested.cc1373 Status IrEmitterUnnested::HandleTupleSelect(HloInstruction* tuple_select) { in HandleTupleSelect() argument
1375 BuildKernelThunk(tuple_select, /*implements_whole_instruction=*/true)); in HandleTupleSelect()
1376 return IrEmitter::HandleTupleSelect(tuple_select); in HandleTupleSelect()
/external/tensorflow/tensorflow/compiler/xla/service/
Dlogical_buffer_analysis.cc168 Status LogicalBufferAnalysis::HandleTupleSelect(HloInstruction* tuple_select) { in HandleTupleSelect() argument
171 NewLogicalBuffer(tuple_select, /*index=*/{}); in HandleTupleSelect()
Dtuple_points_to_analysis.cc395 Status TuplePointsToAnalysis::HandleTupleSelect(HloInstruction* tuple_select) { in HandleTupleSelect() argument
403 auto on_true = tuple_select->operand(1); in HandleTupleSelect()
404 auto on_false = tuple_select->operand(2); in HandleTupleSelect()
405 PointsToSet& points_to_set = CreateCopiedPointsToSet(tuple_select, on_true); in HandleTupleSelect()
423 logical_buffer_analysis_->GetBuffer(tuple_select, /*index=*/{}), in HandleTupleSelect()
Ddfs_hlo_visitor_with_default.h82 Status HandleTupleSelect(HloInstructionPtr tuple_select) override { in HandleTupleSelect() argument
83 return DefaultAction(tuple_select); in HandleTupleSelect()
Dlogical_buffer_analysis.h66 Status HandleTupleSelect(HloInstruction* tuple_select) override;
Dhlo_evaluator.cc1306 Status HloEvaluator::HandleTupleSelect(HloInstruction* tuple_select) { in HandleTupleSelect() argument
1307 const auto& pred = GetEvaluatedLiteralFor(tuple_select->operand(0)); in HandleTupleSelect()
1308 const auto& on_true = GetEvaluatedLiteralFor(tuple_select->operand(1)); in HandleTupleSelect()
1309 const auto& on_false = GetEvaluatedLiteralFor(tuple_select->operand(2)); in HandleTupleSelect()
1312 evaluated_[tuple_select] = on_true.Clone(); in HandleTupleSelect()
1314 evaluated_[tuple_select] = on_false.Clone(); in HandleTupleSelect()
Dtuple_points_to_analysis.h254 Status HandleTupleSelect(HloInstruction* tuple_select) override;
Dhlo_evaluator.h223 Status HandleTupleSelect(HloInstruction* tuple_select) override;
Dhlo_verifier.h46 Status HandleTupleSelect(HloInstruction* tuple_select) override;
Dhlo_verifier.cc116 Status ShapeVerifier::HandleTupleSelect(HloInstruction* tuple_select) { in HandleTupleSelect() argument
117 return CheckTernaryShape(tuple_select); in HandleTupleSelect()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc311 Status IrEmitter::HandleTupleSelect(HloInstruction* tuple_select) { in HandleTupleSelect() argument
312 auto pred = tuple_select->operand(0); in HandleTupleSelect()
313 auto on_true = tuple_select->operand(1); in HandleTupleSelect()
314 auto on_false = tuple_select->operand(2); in HandleTupleSelect()
317 TF_RET_CHECK(tuple_select->shape().IsTuple()); in HandleTupleSelect()
318 TF_RETURN_IF_ERROR(EmitTargetAddressForOp(tuple_select)); in HandleTupleSelect()
319 llvm_ir::EmitTupleSelect(GetIrArrayFor(tuple_select), GetIrArrayFor(pred), in HandleTupleSelect()
Dir_emitter.h153 Status HandleTupleSelect(HloInstruction* tuple_select) override;