Searched refs:node_to_loc (Results 1 – 7 of 7) sorted by relevance
35 NodeToLocFunc node_to_loc = nullptr; member
92 popts.node_to_loc = SplitByDevice; in Partition()450 popts.node_to_loc = SplitByDevice; in TEST_F()
1014 dstp = opts.node_to_loc(dst); in Partition()1072 GraphDef* src_graph = &(*partitions)[opts.node_to_loc(src)]; in Partition()
32 partition_options.node_to_loc = [](const Node* node) { in PartitionFunctionGraph()
1658 popts.node_to_loc = [](const Node* node) { in CreateGraphs()
168 popts.node_to_loc = SplitByDevice; in InitItem()
1672 popts.node_to_loc = SplitByWorker; in BuildAndRegisterPartitions()