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