Searched defs:retval_shardings (Results 1 – 2 of 2) sorted by relevance
101 std::map<int, xla::OpSharding> retval_shardings; in ComputeArgAndRetvalShardings() local168 const std::map<int, xla::OpSharding>& retval_shardings, in BuildComputation()1328 std::map<int, xla::OpSharding> retval_shardings; in CompileGraph() local
2893 const std::vector<xla::OpSharding>& retval_shardings, in BuildExecuteNodes()