Home
last modified time | relevance | path

Searched defs:sink_node (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Ddisable_intra_op_parallelism_test.cc67 NodeDef *sink_node = in TEST_P() local
138 NodeDef sink_node = in TEST_P() local
182 NodeDef sink_node = in TEST() local
Duse_private_thread_pool_test.cc64 NodeDef *sink_node = in TEST_P() local
134 NodeDef sink_node = in TEST_P() local
178 NodeDef sink_node = in TEST() local
Dauto_shard.cc566 Status ShardByFile(const NodeDef& sink_node, int64_t num_workers, int64_t index, in ShardByFile()
574 Status RewriteRebatchV2ToV1(const NodeDef& sink_node, int64_t num_replicas, in RewriteRebatchV2ToV1()
623 Status ShardByData(const NodeDef& sink_node, int64_t num_workers, int64_t index, in ShardByData()
645 Status ShardByHint(const NodeDef& sink_node, int64_t num_workers, int64_t index, in ShardByHint()
689 NodeDef* sink_node; in OptimizeGraph() local
Duse_private_thread_pool.cc62 NodeDef* sink_node = graph.GetNode(item.fetch.at(0)); in OptimizeAndCollectStats() local
Ddisable_intra_op_parallelism.cc70 NodeDef* sink_node = graph.GetNode(item.fetch.at(0)); in OptimizeAndCollectStats() local
Dgraph_utils_test.cc340 NodeDef* sink_node; in TEST() local
355 NodeDef* sink_node; in TEST() local
368 NodeDef* sink_node; in TEST() local
/external/ComputeLibrary/src/graph/
DGraph.cpp79 std::unique_ptr<INode> &sink_node = _nodes[sink]; in add_connection() local
/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_ops_pass_test.cc254 Node* sink_node = graph->sink_node(); in TEST_F() local
/external/tensorflow/tensorflow/core/graph/
Dgraph.h669 Node* sink_node() const { return FindNodeId(kSinkId); } in sink_node() function
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc137 const SimpleNode* sink_node() const { return nodes_[Graph::kSinkId]; } in sink_node() function in tensorflow::tensorrt::segment::__anon982826700111::SimpleGraph