Home
last modified time | relevance | path

Searched refs:copy_start (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/lib/strings/
Dordered_code.cc166 const char* copy_start = p; in EncodeStringFragment() local
173 AppendBytes(dest, copy_start, p - copy_start - 1); in EncodeStringFragment()
176 copy_start = p; in EncodeStringFragment()
179 AppendBytes(dest, copy_start, p - copy_start - 1); in EncodeStringFragment()
182 copy_start = p; in EncodeStringFragment()
185 if (p > copy_start) { in EncodeStringFragment()
186 AppendBytes(dest, copy_start, p - copy_start); in EncodeStringFragment()
222 const char* copy_start = start; in ReadStringInternal() local
233 AppendBytes(result, copy_start, start - copy_start - 1); in ReadStringInternal()
248 copy_start = start; in ReadStringInternal()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dlogical_buffer_analysis.cc164 Status LogicalBufferAnalysis::HandleCopyStart(HloInstruction* copy_start) { in HandleCopyStart() argument
167 NewLogicalBuffer(copy_start, /*index=*/{}); in HandleCopyStart()
168 NewLogicalBuffer(copy_start, /*index=*/{0}); in HandleCopyStart()
169 NewLogicalBuffer(copy_start, /*index=*/{2}); in HandleCopyStart()
Dtuple_points_to_analysis_test.cc338 auto copy_start = builder.AddInstruction(HloInstruction::CreateCopyStart( in TEST_F() local
343 constant->shape(), HloOpcode::kCopyDone, copy_start)); in TEST_F()
347 EXPECT_FALSE(points_to_analysis_->GetPointsToSet(copy_start).IsAmbiguous()); in TEST_F()
348 EXPECT_TRUE(points_to_analysis_->GetPointsToSet(copy_start).IsDistinct()); in TEST_F()
353 points_to_analysis_->GetPointsToSet(copy_start).element({}), in TEST_F()
354 {copy_start}); in TEST_F()
355 ExpectHasBufferAliases(copy_start, {0}, {{copy_start, {0}}, {copy_done, {}}}); in TEST_F()
356 ExpectHasBufferAliases(constant, {}, {{constant, {}}, {copy_start, {1}}}); in TEST_F()
Dhlo_matchers_test.cc279 auto copy_start = HloInstruction::CreateCopyStart( in TEST_F() local
284 shape_memspace2, HloOpcode::kCopyDone, copy_start.get()); in TEST_F()
288 EXPECT_THAT(Explain(copy_start.get(), op::AsyncCopy(2, 1, op::Parameter(0))), in TEST_F()
Dtuple_points_to_analysis.cc320 Status TuplePointsToAnalysis::HandleCopyStart(HloInstruction* copy_start) { in HandleCopyStart() argument
322 PointsToSet& points_to_set = CreateEmptyPointsToSet(copy_start); in HandleCopyStart()
324 GetPointsToSet(copy_start->operand(0)); in HandleCopyStart()
332 &logical_buffer_analysis_->GetBuffer(copy_start, target_index)); in HandleCopyStart()
Dlogical_buffer_analysis.h65 Status HandleCopyStart(HloInstruction* copy_start) override;
Ddfs_hlo_visitor_with_default.h216 Status HandleCopyStart(HloInstructionPtr copy_start) override { in HandleCopyStart() argument
217 return DefaultAction(copy_start); in HandleCopyStart()
Dhlo_dataflow_analysis.h223 bool UpdateCopyStartValueSet(HloInstruction* copy_start);
Dhlo_dataflow_analysis.cc452 bool HloDataflowAnalysis::UpdateCopyStartValueSet(HloInstruction* copy_start) { in UpdateCopyStartValueSet() argument
453 CHECK_EQ(copy_start->opcode(), HloOpcode::kCopyStart); in UpdateCopyStartValueSet()
456 const HloValueSet& operand_value_set = GetValueSet(copy_start->operand(0)); in UpdateCopyStartValueSet()
457 HloValueSet& value_set = GetValueSet(copy_start, {1}); in UpdateCopyStartValueSet()
Dtuple_points_to_analysis.h253 Status HandleCopyStart(HloInstruction* copy_start) override;
Dhlo_evaluator.h231 Status HandleCopyStart(HloInstruction* copy_start) override;
Dhlo_verifier.h98 Status HandleCopyStart(HloInstruction* copy_start) override;
Dhlo_dataflow_analysis_test.cc1232 auto copy_start = builder.AddInstruction(HloInstruction::CreateCopyStart( in TEST_P() local
1237 constant->shape(), HloOpcode::kCopyDone, copy_start)); in TEST_P()
1246 EXPECT_TRUE(analysis.ValueIsDefinedAt(copy_start, /*index=*/{})); in TEST_P()
1247 EXPECT_TRUE(analysis.ValueIsDefinedAt(copy_start, /*index=*/{0})); in TEST_P()
1248 EXPECT_FALSE(analysis.ValueIsDefinedAt(copy_start, /*index=*/{1})); in TEST_P()
1249 EXPECT_TRUE(analysis.ValueIsDefinedAt(copy_start, /*index=*/{2})); in TEST_P()
1253 UnorderedElementsAre(analysis.GetValueDefinedAt(copy_start, {0}))); in TEST_P()
1254 EXPECT_TRUE(analysis.GetValueDefinedAt(copy_start, /*index=*/{0}) in TEST_P()
Dhlo_evaluator.cc1879 Status HloEvaluator::HandleCopyStart(HloInstruction* copy_start) { in HandleCopyStart() argument
1880 if (copy_start->user_count() != 1 || in HandleCopyStart()
1881 copy_start->users().at(0)->opcode() != HloOpcode::kCopyDone) { in HandleCopyStart()
1893 evaluated_[copy_start] = LiteralUtil::MakeTuple( in HandleCopyStart()
1894 {&GetEvaluatedLiteralFor(copy_start->operand(0)), in HandleCopyStart()
1895 &GetEvaluatedLiteralFor(copy_start->operand(0)), &context_literal}); in HandleCopyStart()
Dhlo_verifier.cc983 Status ShapeVerifier::HandleCopyStart(HloInstruction* copy_start) { in HandleCopyStart() argument
984 return CheckShape(copy_start, in HandleCopyStart()
985 ShapeUtil::MakeTupleShape({copy_start->operand(0)->shape(), in HandleCopyStart()
986 copy_start->operand(0)->shape(), in HandleCopyStart()
Ddfs_hlo_visitor.h268 virtual Status HandleCopyStart(HloInstructionPtr copy_start) = 0;
Dmemory_space_assignment.cc721 copy_start() == other.copy_start() && copy_done() == other.copy_done(); in operator ==()
3164 << copy_allocation->copy_start()->ToString() in ScheduleAsynchronousCopies()
3171 copy_allocation->copy_start()); in ScheduleAsynchronousCopies()
Dmemory_space_assignment.h617 HloInstruction* copy_start() const { return copy_start_; } in copy_start() function
Dmemory_space_assignment_test.cc1460 const HloInstruction* copy_start = nullptr; in TEST_P() local
1463 copy_start = user; in TEST_P()
1467 EXPECT_NE(copy_start, nullptr); in TEST_P()
1469 ShapeUtil::GetSubshape(copy_start->shape(), {0}); in TEST_P()