Searched refs:node_to_loc (Results 1 – 8 of 8) sorted by relevance
40 std::function<string(const Node*)> node_to_loc, in PartitionFunctionGraph() argument43 if (node_to_loc != nullptr) { in PartitionFunctionGraph()44 partition_options.node_to_loc = node_to_loc; in PartitionFunctionGraph()46 partition_options.node_to_loc = [](const Node* node) { in PartitionFunctionGraph()105 auto node_to_loc = [](const Node* node) { in InsertTransferOps() local112 if (*location != node_to_loc(node)) { in InsertTransferOps()117 location = node_to_loc(node); in InsertTransferOps()130 &partitions, node_to_loc, in InsertTransferOps()
1675 popts.node_to_loc = [](const Node* node) { in CreateGraphs()
35 NodeToLocFunc node_to_loc = nullptr; member
93 popts.node_to_loc = SplitByDevice; in Partition()451 popts.node_to_loc = SplitByDevice; in TEST_F()
1015 dstp = opts.node_to_loc(dst); in Partition()1073 GraphDef* src_graph = &(*partitions)[opts.node_to_loc(src)]; in Partition()
170 popts.node_to_loc = SplitByDevice; in InitItem()
1690 popts.node_to_loc = SplitByWorker; in BuildAndRegisterPartitions()
2416 partition_options.node_to_loc = [](const Node* node) { in PartitionHelper()