Home
last modified time | relevance | path

Searched refs:hlo_ordering (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.h463 const HloOrdering& hlo_ordering() const { return *hlo_ordering_; } in hlo_ordering() function
498 std::unique_ptr<HloOrdering> hlo_ordering, in BufferAssignment() argument
504 hlo_ordering_(std::move(hlo_ordering)), in BufferAssignment()
613 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering,
634 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering,
Dbuffer_assignment.cc729 hlo_ordering().SequentialOrder(*computation); in ComputeSummaryStats()
910 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering, in Run() argument
920 module, std::move(hlo_ordering), std::move(buffer_size), in Run()
1058 } else if (assignment->hlo_ordering().MayInterfere( in MaybeAssignBuffer()
1202 assignment->hlo_ordering().SequentialOrder(*computation) != nullptr; in AssignSingleHloBuffer()
1295 assignment->hlo_ordering().SequentialOrder(*computation); in AssignBuffersForComputations()
1414 const HloOrdering& hlo_ordering = assignment->hlo_ordering(); in AssignBuffersWithSequentialOrdering() local
1445 hlo_ordering.SequentialOrder(*computation); in AssignBuffersWithSequentialOrdering()
1477 hlo_ordering.SequentialOrder(*computation); in AssignBuffersWithSequentialOrdering()
1626 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering, in CreateAssignment() argument
[all …]
DBUILD332 ":hlo_ordering",
355 ":hlo_ordering",
585 ":hlo_ordering",
1352 ":hlo_ordering",
1372 name = "hlo_ordering",
1373 srcs = ["hlo_ordering.cc"],
1374 hdrs = ["hlo_ordering.h"],
1402 ":hlo_ordering",
1424 ":hlo_ordering",
1446 ":hlo_ordering",
[all …]
Dhlo.proto509 reserved "hlo_ordering";
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
DBUILD202 "//tensorflow/compiler/xla/service:hlo_ordering",
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc1543 assignment.hlo_ordering().SequentialOrder(*computation); in HloToLhloModule()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
DBUILD1556 "//tensorflow/compiler/xla/service:hlo_ordering",