Home
last modified time | relevance | path

Searched defs:retval_shapes (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.cc1908 std::vector<InferredShape>* retval_shapes) { in GetArgAndRetvalShapes()
2129 const std::vector<InferredShape>& retval_shapes, const Graph& graph, in AssignArgsAndRetvalsToCores()
4754 std::vector<InferredShape> arg_shapes, retval_shapes; in RewriteTPUReplicateNode() local