Searched defs:retval_sharding (Results 1 – 2 of 2) sorted by relevance
285 if (auto retval_sharding = GetXlaShardingFromRetval(retval.get())) { in IdentifyXlaShardingForComputationOutputs() local
1920 std::vector<xla::OpSharding>* retval_sharding, in AssignArgsAndRetvalsToCores()2314 const std::vector<xla::OpSharding>& retval_sharding, in BuildCompileNode()4261 std::vector<xla::OpSharding> retval_sharding; in RewriteTPUReplicateNode() local