Searched defs:retval_shapes (Results 1 – 1 of 1) sorted by relevance
1695 std::vector<InferredShape>* retval_shapes) { in GetArgAndRetvalShapes()1916 const std::vector<InferredShape>& retval_shapes, const Graph& graph, in AssignArgsAndRetvalsToCores()4252 std::vector<InferredShape> arg_shapes, retval_shapes; in RewriteTPUReplicateNode() local