- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for MergeInferredShapes (0.71 sec)
-
tensorflow/compiler/jit/shape_inference.h
GraphShapeInfo* shape_info); // Merges two InferredShapes. Return an error if the two shapes cannot be // merged. absl::StatusOr<InferredShape> MergeInferredShapes(const InferredShape& a, const InferredShape& b); } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/jit/shape_inference.cc
// backward information flow is added in the future. return StoreOutputShapes(*graph, shape_refiner, shape_info); } absl::StatusOr<InferredShape> MergeInferredShapes(const InferredShape& a, const InferredShape& b) { InferredShape result; TF_RETURN_IF_ERROR(a.shape.MergeWith(b.shape, &result.shape));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 13K bytes - Viewed (0)