Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dlogical_buffer_analysis.cc102 Status LogicalBufferAnalysis::DefaultAction(HloInstruction* hlo_instruction) { in DefaultAction() argument
105 hlo_instruction->shape(), in DefaultAction()
106 [this, hlo_instruction](const Shape& shape, const ShapeIndex& index) { in DefaultAction()
107 NewLogicalBuffer(hlo_instruction, index); in DefaultAction()
Dtuple_points_to_analysis.cc201 Status TuplePointsToAnalysis::DefaultAction(HloInstruction* hlo_instruction) { in DefaultAction() argument
205 PointsToSet& points_to_set = CreateEmptyPointsToSet(hlo_instruction); in DefaultAction()
207 [this, hlo_instruction](const ShapeIndex& index, in DefaultAction()
210 &logical_buffer_analysis_->GetBuffer(hlo_instruction, index)); in DefaultAction()
213 if (hlo_instruction->shape().IsTuple()) { in DefaultAction()
216 points_to_set.add_tuple_source({}, hlo_instruction); in DefaultAction()
509 const HloInstruction* hlo_instruction) const { in GetPointsToSet()
510 return *PerInst(hlo_instruction)->points_to_set; in GetPointsToSet()
Ddfs_hlo_visitor_with_default.h53 virtual Status DefaultAction(HloInstructionPtr hlo_instruction) = 0;
338 Status DefaultAction(HloInstructionPtr hlo_instruction) override { in DefaultAction() argument
339 return visitor_func_(hlo_instruction); in DefaultAction()
Dtuple_points_to_analysis.h193 const HloInstruction* hlo_instruction) const;
247 Status DefaultAction(HloInstruction* hlo_instruction) override;
Dlogical_buffer_analysis.h59 Status DefaultAction(HloInstruction* hlo_instruction) override;
Dhlo_computation_test.cc234 Status DefaultAction(HloInstruction* hlo_instruction) override { in TEST_F() argument
235 EXPECT_FALSE(visited_set_.contains(hlo_instruction)); in TEST_F()
236 visited_set_.insert(hlo_instruction); in TEST_F()
237 last_visited_ = hlo_instruction; in TEST_F()
Dhlo_cost_analysis.h211 Status HandleElementwiseOp(const HloInstruction* hlo_instruction);
Dhlo_cost_analysis.cc94 const HloInstruction* hlo_instruction) { in HandleElementwiseOp() argument
95 const auto& shape = hlo_instruction->shape(); in HandleElementwiseOp()
99 auto opcode = hlo_instruction->opcode(); in HandleElementwiseOp()
Dhlo_instruction_test.cc55 Status DefaultAction(HloInstruction* hlo_instruction) override { in DefaultAction() argument
57 HloOpcodeString(hlo_instruction->opcode())); in DefaultAction()
577 Status DefaultAction(HloInstruction* hlo_instruction) override { in DefaultAction() argument
578 visited_nodes_.push_back(hlo_instruction); in DefaultAction()
Dhlo_evaluator_typed_visitor.h180 Status DefaultAction(HloInstruction* hlo_instruction) override {
182 HloOpcodeString(hlo_instruction->opcode()));
DBUILD414 "hlo_instruction.cc",
432 "hlo_instruction.h",
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc242 Status DefaultAction(HloInstruction* hlo_instruction) override { in DefaultAction() argument
244 {hlo_instruction, FindOrDie(assigned_indices_, hlo_instruction)}); in DefaultAction()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md1868 …reateRecvDone`](https://www.tensorflow.org/code/tensorflow/compiler/xla/service/hlo_instruction.h).
2651 …reateSendDone`](https://www.tensorflow.org/code/tensorflow/compiler/xla/service/hlo_instruction.h).