Home
last modified time | relevance | path

Searched refs:ReplicatePerReplicaNodesInFunctionGraph (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dreplicate_per_replica_nodes_test.cc109 ReplicatePerReplicaNodesInFunctionGraph(composite_devices, &graph)); in TEST()
150 ReplicatePerReplicaNodesInFunctionGraph(composite_devices, &graph)); in TEST()
201 ReplicatePerReplicaNodesInFunctionGraph(composite_devices, &graph)); in TEST()
268 ReplicatePerReplicaNodesInFunctionGraph(composite_devices, &graph)); in TEST()
315 ReplicatePerReplicaNodesInFunctionGraph(composite_devices, &graph)); in TEST()
H A Dreplicate_per_replica_nodes.h37 Status ReplicatePerReplicaNodesInFunctionGraph(
H A Dreplicate_per_replica_nodes.cc248 Status ReplicatePerReplicaNodesInFunctionGraph( in ReplicatePerReplicaNodesInFunctionGraph() function
H A Dprocess_function_library_runtime.cc958 TF_RETURN_IF_ERROR(ReplicatePerReplicaNodesInFunctionGraph( in InstantiateMultiDevice()