| /external/tensorflow/tensorflow/compiler/xla/service/ |
| tuple_points_to_analysis_test.cc | 57 points_to_analysis_ =
66 points_to_analysis_->GetPointsToSet(instruction).element(index);
110 points_to_analysis_->GetBufferDefinedAt(instruction, index)
116 EXPECT_THAT(points_to_analysis_->GetBufferAliases(*buffer),
121 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_; member in class:xla::__anon39011::TuplePointsToAnalysisTest
134 EXPECT_EQ(1, points_to_analysis_->GetPointsToSet(constant1).size());
136 points_to_analysis_->GetPointsToSet(constant1).element({}), {constant1});
138 points_to_analysis_->GetPointsToSet(constant1).tuple_sources({}).empty());
139 EXPECT_TRUE(points_to_analysis_->GetPointsToSet(tuple).IsDistinct());
141 EXPECT_EQ(1, points_to_analysis_->GetPointsToSet(constant2).size())
[all...] |
| buffer_liveness.cc | 48 TF_ASSIGN_OR_RETURN(points_to_analysis_, TuplePointsToAnalysis::Run(module_));
58 points_to_analysis_->GetPointsToSet(instruction)
69 points_to_analysis_->GetPointsToSet(root).CreateFlattenedSet();
98 TF_CHECK_OK(points_to_analysis_->VerifyBuffer(a));
99 TF_CHECK_OK(points_to_analysis_->VerifyBuffer(b));
105 for (const BufferAlias& alias : points_to_analysis_->GetBufferAliases(a)) {
133 for (const BufferAlias& alias : points_to_analysis_->GetBufferAliases(a)) {
173 TF_CHECK_OK(points_to_analysis_->VerifyBuffer(buffer));
|
| liveness_util_test.cc | 36 points_to_analysis_ =
48 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_; member in class:xla::__anon38992::PointsToAnalysisTestBase
71 EXPECT_TRUE(DoesNotUseOperandBuffer(tuple, {0}, gte0, *points_to_analysis_));
72 EXPECT_TRUE(DoesNotUseOperandBuffer(tuple, {1}, gte1, *points_to_analysis_));
73 EXPECT_FALSE(DoesNotUseOperandBuffer(tuple, {}, gte0, *points_to_analysis_));
74 EXPECT_FALSE(DoesNotUseOperandBuffer(tuple, {}, gte1, *points_to_analysis_));
112 DoesNotUseOperandBuffer(tuple, {0}, fusion, *points_to_analysis_));
114 DoesNotUseOperandBuffer(tuple, {1}, fusion, *points_to_analysis_));
137 CanShareOperandBufferWithUser(param, {}, exp, {}, *points_to_analysis_));
139 CanShareOperandBufferWithUser(exp, {}, log, {}, *points_to_analysis_));
[all...] |
| buffer_liveness.h | 63 return *points_to_analysis_;
110 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_;
|
| hlo_rematerialization.h | 126 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_; member in class:xla::HloRematerialization
|
| layout_assignment.h | 137 return points_to_analysis_;
219 const TuplePointsToAnalysis& points_to_analysis_; member in class:xla::LayoutConstraints
|
| hlo_scheduling.cc | 97 points_to_analysis_(points_to_analysis),
168 points_to_analysis_.GetBuffersDefinedByInstruction(instruction)) {
315 const TuplePointsToAnalysis& points_to_analysis_; member in class:xla::__anon38971::ListScheduler
|
| heap_simulator_test.cc | 81 points_to_analysis_ =
92 instruction_sequence, *points_to_analysis_, zero_size)
104 points_to_analysis_ =
126 module_sequence, *points_to_analysis_, size_fn)
135 return points_to_analysis_->GetBufferDefinedAt(instruction, index)
158 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_; member in class:xla::__anon38934::HeapSimulatorTracker
[all...] |
| layout_assignment.cc | 182 : points_to_analysis_(points_to_analysis), computation_(computation) {
184 for (LogicalBuffer::Id id = 0; id < points_to_analysis_.num_logical_buffers();
186 auto& buffer = points_to_analysis_.logical_buffer(id);
200 points_to_analysis_.GetPointsToSet(instruction).CreateFlattenedSet();
202 points_to_analysis_.GetPointsToSet(instruction->operand(operand_no))
218 TF_RETURN_IF_ERROR(points_to_analysis_.VerifyBuffer(buffer));
379 points_to_analysis_.GetPointsToSet(instruction).element(index);
439 points_to_analysis_.GetBuffersDefinedByInstruction(instruction)) {
[all...] |
| hlo_rematerialization.cc | [all...] |
