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