Searched refs:ReadyNodeManager (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | virtual_scheduler.h | 161 class ReadyNodeManager { 163 ReadyNodeManager() {} in ReadyNodeManager() function 164 virtual ~ReadyNodeManager() {} in ~ReadyNodeManager() 175 class FIFOManager : public ReadyNodeManager { 177 FIFOManager() : ReadyNodeManager() {} in FIFOManager() 195 class LIFOManager : public ReadyNodeManager { 197 LIFOManager() : ReadyNodeManager() {} in LIFOManager() 216 class HeapReadyManager : public ReadyNodeManager { 284 class CompositeNodeManager : public ReadyNodeManager { 315 std::unique_ptr<ReadyNodeManager> ReadyNodeManagerFactory( [all …]
|
D | analytical_cost_estimator.h | 47 std::unique_ptr<ReadyNodeManager> node_manager, 52 std::unique_ptr<ReadyNodeManager> node_manager, 72 std::unique_ptr<ReadyNodeManager> node_manager_;
|
D | analytical_cost_estimator.cc | 39 void AddCostNode(ReadyNodeManager* node_manager, const OpContext& op_context, in AddCostNode() 120 std::unique_ptr<ReadyNodeManager> node_manager, bool use_static_shapes, in AnalyticalCostEstimator() 134 std::unique_ptr<ReadyNodeManager> node_manager, in AnalyticalCostEstimator()
|
D | virtual_scheduler.cc | 161 HeapReadyManager::HeapReadyManager() : ReadyNodeManager() { in HeapReadyManager() 269 : ReadyNodeManager(), send_manager_(), recv_manager_() {} in CompositeNodeManager() 367 std::unique_ptr<ReadyNodeManager> ReadyNodeManagerFactory( in ReadyNodeManagerFactory() 1318 ReadyNodeManager* ready_nodes, in VirtualScheduler() 1326 ReadyNodeManager* ready_nodes, in VirtualScheduler()
|
D | virtual_scheduler_test.cc | 649 ReadyNodeManager* ready_node_manager, Cluster* cluster) in TestVirtualScheduler()
|
/external/tensorflow/tensorflow/core/grappler/clusters/ |
D | virtual_cluster.h | 40 std::unique_ptr<ReadyNodeManager> node_manager);
|
D | virtual_cluster.cc | 35 std::unique_ptr<ReadyNodeManager> node_manager) in VirtualCluster()
|