Searched refs:GetEdgeShape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 1679 static Status GetEdgeShape(const GraphShapeInfo& shape_info, const Edge& edge, in GetEdgeShape() function
1712 TF_RETURN_IF_ERROR(GetEdgeShape(shape_info, *input_edges[edge_pos], &info)); in GetArgAndRetvalShapes()
1755 TF_RETURN_IF_ERROR(GetEdgeShape(shape_info, *input_edges[edge_pos], &info)); in GetArgAndRetvalShapes()
1767 TF_RETURN_IF_ERROR(GetEdgeShape(shape_info, *input_edges[edge_pos], &info)); in GetArgAndRetvalShapes()
1787 TF_RETURN_IF_ERROR(GetEdgeShape(shape_info, *input_edges[edge_pos], &info)); in GetArgAndRetvalShapes()
|