Home
last modified time | relevance | path

Searched refs:GetComputationForTPUReplicateOp (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.h287 static Status GetComputationForTPUReplicateOp(const NameAttrList& function,
H A Ddistributed_tpu_rewrite_pass.cc1874 Status DistributedTPURewritePass::GetComputationForTPUReplicateOp( in GetComputationForTPUReplicateOp() function in tensorflow::DistributedTPURewritePass
4463 TF_RETURN_IF_ERROR(GetComputationForTPUReplicateOp( in GetIOTypes()