Searched refs:GetIOTypes (Results 1 – 2 of 2) sorted by relevance
528 static Status GetIOTypes(
3971 /* static */ Status DistributedTPURewritePass::GetIOTypes( in GetIOTypes() function in tensorflow::DistributedTPURewritePass4248 TF_RETURN_IF_ERROR(GetIOTypes(num_replicas, *replicate_node, flr, graph, in RewriteTPUReplicateNode()