Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for RewriteOutsideCompilationSubgraphFn (0.3 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc 

      EXPECT_NE(add_node, nullptr);
  add_node->AddAttr(kXlaConnectedToXlaComputationAttrName, "cluster");
  add_node->AddAttr(kXlaConnectedFromXlaComputationAttrName, "cluster");

  RewriteOutsideCompilationSubgraphFn rewrite_fn("_xla", "_oc", "cluster", "");
  std::vector<OutputTensor> arg_source_tensors;
  NodeDef call_node_def;
  call_node_def.set_op("0");
  TF_CHECK_OK(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 41K bytes
    - Viewed (0)
Back to top