Searched refs:PartitionFunctionGraph (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | partitioning_utils.cc | 37 Status PartitionFunctionGraph( in PartitionFunctionGraph() function
75 Status PartitionFunctionGraph( in PartitionFunctionGraph() function
81 PartitionFunctionGraph(device_set, graph.get(), &partitions, in PartitionFunctionGraph()
129 TF_RETURN_IF_ERROR(PartitionFunctionGraph(device_set, graph.get(), in InsertTransferOps()
|
| H A D | partitioning_utils_test.cc | 127 PartitionFunctionGraph(device_set_, std::move(graph), &subgraphs); in TEST_F()
138 PartitionFunctionGraph(device_set_, std::move(graph), &subgraphs); in TEST_F()
153 PartitionFunctionGraph(device_set_, std::move(graph), &subgraphs); in TEST_F()
|
| H A D | partitioning_utils.h | 36 Status PartitionFunctionGraph(
|
| H A D | process_function_library_runtime.cc | 973 PartitionFunctionGraph(*dev_set, std::move(graph), &subgraphs)); in InstantiateMultiDevice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/ |
| H A D | graph_partition.cc | 438 PartitionFunctionGraph(device_set, std::move(graph), &partitions)); in InsertTransferOps()
|