Home
last modified time | relevance | path

Searched refs:node_to_loc (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.h35 NodeToLocFunc node_to_loc = nullptr; member
Dgraph_partition_test.cc92 popts.node_to_loc = SplitByDevice; in Partition()
450 popts.node_to_loc = SplitByDevice; in TEST_F()
Dgraph_partition.cc987 dstp = opts.node_to_loc(dst); in Partition()
1045 GraphDef* src_graph = &(*partitions)[opts.node_to_loc(src)]; in Partition()
/external/tensorflow/tensorflow/core/common_runtime/
Dpartitioning_utils.cc29 partition_options.node_to_loc = [](const Node* node) { in PartitionFunctionGraph()
Ddirect_session.cc1534 popts.node_to_loc = [](const Node* node) { in CreateGraphs()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc156 popts.node_to_loc = SplitByDevice; in InitItem()
Dmaster_session.cc1580 popts.node_to_loc = SplitByWorker; in BuildAndRegisterPartitions()