- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for PerformStaticShapeInferenceBeforeEncapsulation (0.51 sec)
-
tensorflow/compiler/jit/encapsulate_util.h
// encapsulation, some nodes will be encapsulated into function call, and shape // inference does not handle function call at the moment. Status PerformStaticShapeInferenceBeforeEncapsulation(Graph* g); // Attribute indicating that some ops in this node's XLA computation has control // dependency on this node. Attribute value will always be "true".
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_util.cc
const char kXlaOutsideCompilationInputsAttrName[] = "_xla_oc_inputs"; 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(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.5K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc
options.allow_internal_ops = true; std::unique_ptr<Graph> graph(new Graph(lib_def.get())); s = ConvertGraphDefToGraph(options, *graphdef, graph.get()); if (!s.ok()) return s; s = PerformStaticShapeInferenceBeforeEncapsulation(graph.get()); if (!s.ok()) return s; // Create FunctionLibraryRuntime. SessionOptions session_options; std::vector<std::unique_ptr<Device>> devices;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0)