Home
last modified time | relevance | path

Searched defs:name_to_node (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dutils.cc83 const std::unordered_map<string, const NodeDef*>& name_to_node, in ExtractExtraProperties()
167 const std::unordered_map<string, const NodeDef*>& name_to_node) { in FindInputFeatures()
273 const std::unordered_map<string, const NodeDef*>& name_to_node, in BuildOpInfoWithoutDevice()
300 std::unordered_map<string, const NodeDef*> name_to_node; in CostGraphToOpPerformanceData() local
Dutils_test.cc76 std::unordered_map<string, const NodeDef*> name_to_node; in TEST() local
164 std::unordered_map<string, const NodeDef*> name_to_node; in TEST() local
Dvirtual_scheduler.cc447 std::unordered_map<string, const NodeDef*> name_to_node; in Init() local
Dgraph_properties.cc2792 std::unordered_map<string, const NodeDef*> name_to_node; // Empty in InferFromCostGraph() local
/external/tensorflow/tensorflow/core/grappler/utils/
Dscc_test.cc86 std::unordered_map<string, const NodeDef*> name_to_node; in TEST_F() local
149 std::unordered_map<string, const NodeDef*> name_to_node; in TEST_F() local
Dtransitive_fanin.cc32 std::unordered_map<string, const NodeDef*> name_to_node; in ComputeTransitiveFanin() local
/external/tensorflow/tensorflow/python/framework/
Dgraph_util_impl.py201 def _assert_nodes_are_present(name_to_node, nodes): argument
/external/tensorflow/tensorflow/compiler/jit/
Dintroduce_floating_point_jitter_pass.cc29 absl::flat_hash_map<string, Node*> name_to_node; in GetNodesToModify() local
/external/tensorflow/tensorflow/core/tfrt/utils/
Dtfrt_graph_execution_state.cc507 const absl::flat_hash_map<std::string, NodeDef*>& name_to_node) { in FindLoopCondFromExitNode()
549 absl::flat_hash_map<std::string, NodeDef*> name_to_node; in PruneGraphDef() local
Dtfrt_graph_execution_state_test.cc146 absl::flat_hash_map<std::string, NodeDef> name_to_node; in TEST_F() local
208 absl::flat_hash_map<std::string, NodeDef> name_to_node; in TEST_F() local
275 absl::flat_hash_map<std::string, NodeDef> name_to_node; in TEST_F() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmake_deterministic.cc154 absl::flat_hash_map<absl::string_view, const NodeDef*> name_to_node; in NameToNode() local
224 absl::flat_hash_map<absl::string_view, const NodeDef*> name_to_node = in GetAllTransitiveDependencies() local
Dsplit_utils.cc42 absl::flat_hash_map<absl::string_view, const NodeDef*> name_to_node; in NameToNode() local
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session.h166 NameNodeMap name_to_node; member
Dstep_stats_collector.cc318 std::unordered_map<StringPiece, Node*, StringPieceHasher> name_to_node; in BuildCostModel() local
Ddirect_session.cc1252 const NameNodeMap* name_to_node = &executors_and_keys->name_to_node; in CheckFetch() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc528 std::unordered_map<string, const NodeDef*> name_to_node; in AddPlaceholdersForFeeds() local