Searched refs:infer_from_computation (Results 1 – 1 of 1) sorted by relevance
183 bool infer_from_computation, tf_device::ClusterFuncOp cluster_func, in IdentifyXlaShardingForComputationInputs() argument218 if (infer_from_computation) { in IdentifyXlaShardingForComputationInputs()353 bool infer_from_computation, tf_device::ClusterFuncOp cluster_func, in IdentifyXlaShardingForComputationOutputs() argument389 if (infer_from_computation) { in IdentifyXlaShardingForComputationOutputs()