| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| H A D | auto_shard_test.cc | 96 NodeDef* sink_node = nullptr; in TEST() local
128 NodeDef* sink_node = nullptr; in TEST() local
161 NodeDef* sink_node = nullptr; in TEST() local
187 NodeDef* sink_node = nullptr; in TEST() local
213 NodeDef* sink_node = nullptr; in TEST() local
240 NodeDef* sink_node = nullptr; in TEST() local
266 NodeDef* sink_node = nullptr; in TEST() local
290 NodeDef* sink_node = nullptr; in TEST() local
319 NodeDef* sink_node = nullptr; in TEST() local
342 NodeDef* sink_node = nullptr; in TEST() local
[all …]
|
| H A D | auto_shard.cc | 647 Status ShardByFile(const NodeDef& sink_node, int64_t num_workers, int64_t index, in ShardByFile()
655 Status RewriteRebatchV2ToV1(const NodeDef& sink_node, int64_t num_replicas, in RewriteRebatchV2ToV1()
704 Status ShardByData(const NodeDef& sink_node, int64_t num_workers, int64_t index, in ShardByData()
726 Status ShardByHint(const NodeDef& sink_node, int64_t num_workers, int64_t index, in ShardByHint()
756 Status ApplyAutoShard(const NodeDef& sink_node, int64_t num_workers, in ApplyAutoShard()
793 NodeDef* sink_node; in OptimizeGraph() local
825 bool IsEligibleRewriteBatchSize(const NodeDef& sink_node, in IsEligibleRewriteBatchSize()
|
| H A D | disable_intra_op_parallelism_test.cc | 67 NodeDef *sink_node = in TEST_P() local
138 NodeDef sink_node = in TEST_P() local
182 NodeDef sink_node = in TEST() local
|
| H A D | use_private_thread_pool_test.cc | 64 NodeDef *sink_node = in TEST_P() local
134 NodeDef sink_node = in TEST_P() local
178 NodeDef sink_node = in TEST() local
|
| H A D | graph_utils_test.cc | 347 NodeDef* sink_node; in TEST() local
362 NodeDef* sink_node; in TEST() local
375 NodeDef* sink_node; in TEST() local
|
| H A D | use_private_thread_pool.cc | 61 NodeDef* sink_node = graph.GetNode(item.fetch.at(0)); in OptimizeAndCollectStats() local
|
| H A D | disable_intra_op_parallelism.cc | 69 NodeDef* sink_node = graph.GetNode(item.fetch.at(0)); in OptimizeAndCollectStats() local
|
| H A D | inject_prefetch.cc | 104 NodeDef* sink_node = graph.GetNode(item.fetch.at(0)); in OptimizeAndCollectStats() local
|
| /aosp_15_r20/external/ComputeLibrary/src/graph/ |
| H A D | Graph.cpp | 79 std::unique_ptr<INode> &sink_node = _nodes[sink]; in add_connection() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | build_xla_ops_pass_test.cc | 253 Node* sink_node = graph->sink_node(); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | graph.h | 686 Node* sink_node() const { return FindNodeId(kSinkId); } in sink_node() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| H A D | segment.cc | 141 const SimpleNode* sink_node() const { return nodes_[Graph::kSinkId]; } in sink_node() function in tensorflow::tensorrt::segment::__anon2febd5fa0111::SimpleGraph
|