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