- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for InferredShape (0.14 sec)
-
tensorflow/compiler/jit/shape_inference.h
Status InferShapes(Graph* graph, const std::map<int, InferredShape>& arg_shapes, const tensorflow::FunctionLibraryDefinition* fnlib_def, 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);
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)); if (a.handle_type == DT_INVALID) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 13K bytes - Viewed (0) -
tensorflow/compiler/jit/shape_inference_test.cc
a.node()->AddAttr("_index", 0); b.node()->AddAttr("_index", 1); std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); TF_CHECK_OK(root.ToGraph(graph.get())); std::map<int, InferredShape> arg_shapes; arg_shapes[0].shape = TensorShape({2, 3}); arg_shapes[1].shape = TensorShape({2, 3}); GraphShapeInfo shape_info; TF_ASSERT_OK(InferShapes(graph.get(), arg_shapes,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_util.cc
const char kXlaIsPlaceholderForArg[] = "_xla_is_placeholder_for_arg"; Status PerformStaticShapeInferenceBeforeEncapsulation(Graph* g) { // Perform shape inference. std::map<int, InferredShape> arg_shapes; GraphShapeInfo shape_info; TF_RETURN_IF_ERROR( InferShapes(g, arg_shapes, /*fnlib_def=*/nullptr, &shape_info)); // Add attribute for output shapes.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.5K bytes - Viewed (0)