Home
last modified time | relevance | path

Searched refs:WindowedDotGeneralLoop (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h504 struct WindowedDotGeneralLoop { struct
541 std::vector<WindowedDotGeneralLoop> windowed_dot_general_loops_;
Ddot_handler.cc481 std::vector<SpmdPartitioningVisitor::WindowedDotGeneralLoop>* in PartitionBaseCase()
1645 std::vector<SpmdPartitioningVisitor::WindowedDotGeneralLoop>*
1660 std::vector<SpmdPartitioningVisitor::WindowedDotGeneralLoop>* in PartitionDotGroupOnBatch()
1994 std::vector<SpmdPartitioningVisitor::WindowedDotGeneralLoop>* in PartitionDotGroupOnNonContracting()
2089 std::vector<SpmdPartitioningVisitor::WindowedDotGeneralLoop>* in PartitionDotGroupOnContracting()
2441 std::vector<SpmdPartitioningVisitor::WindowedDotGeneralLoop>* in PartitionDot()
2807 std::vector<SpmdPartitioningVisitor::WindowedDotGeneralLoop>* in PartitionDot()