Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for AddFunctionLibrary (0.18 sec)

  1. tensorflow/compiler/jit/build_xla_ops_pass_test.cc

      Scope root = Scope::NewRootScope().WithDevice(kXlaDeviceName).ExitOnError();
    
      FunctionDefLibrary fdef_lib =
          CreateFunctionDefLibWithConstFunction("cluster_0");
      TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib));
      Node* call;
      TF_ASSERT_OK(MakeXlaCompiledKernel(root.graph(), "cluster_0", "C", &call));
      call->AddAttr(kXlaHasReferenceVarsAttr, false);
      call->set_requested_device(kXlaDeviceName);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 12.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/compilability_check_util_test.cc

      NameAttrList uncompilable_fn2_attr;
      uncompilable_fn2_attr.set_name(kUncompilableFunctionTwoName);
    
      Scope root = Scope::NewRootScope().ExitOnError();
      TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib));
      auto predicate = ops::Placeholder(root.WithOpName("pred"), DT_BOOL);
      auto placeholder = ops::Placeholder(root.WithOpName("A"), DT_INT32);
      std::vector<NodeBuilder::NodeOut> if_inputs(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 10 12:32:39 UTC 2022
    - 22.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/resource_operation_safety_analysis_test.cc

    TEST(ResourceOperationSafetyAnalysisTest, CallRead) {
      Scope root = Scope::NewRootScope().ExitOnError();
    
      FunctionDefLibrary flib_def =
          CreateFunctionDefLibWithConstFunction("Const_func");
      TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def));
    
      Node* read = MakeRead(root, "R");
      Status status;
      Node* call = MakeCall(root.graph(), "Const_func", "C", &status);
      TF_ASSERT_OK(status);
    
      root.graph()->AddControlEdge(call, read);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 28 16:53:59 UTC 2020
    - 18.7K bytes
    - Viewed (0)
  4. tensorflow/cc/gradients/functional_grad_test.cc

      // Construct a graph:
      //   A = Placeholder[dtype=int32]
      //   B = XTimesTwo[_tpu_replicate="cluster"](A)
      //   C = XTimesTwo[_xla_compile_id="cluster"](A)
      TF_ASSERT_OK(scope_.graph()->AddFunctionLibrary(f_lib_proto));
    
      Output x = Placeholder(scope_, DT_FLOAT);
      NameAttrList f;
      f.set_name("XTimesTwo");
      (*f.mutable_attr())["T"].set_type(DT_FLOAT);
      auto results =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 26 14:42:24 UTC 2021
    - 3K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/force_xla_constants_on_host_pass_test.cc

      AttrValue true_attribute;
      true_attribute.set_b(true);
      (*called_func.mutable_attr())[kXlaMustCompileAttr] = true_attribute;
      *library.add_function() = called_func;
      TF_ASSERT_OK(root.graph()->AddFunctionLibrary(library));
      FunctionLibraryDefinition flib_def(OpRegistry::Global(), library);
      Output in = ops::Placeholder(root, DT_FLOAT);
      Output perm = ops::Const(root, {3, 1, 2, 0});
    
      NameAttrList b_name_attr;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 4.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/encapsulate_xla_computations_pass_test.cc

    namespace tensorflow {
    
    static std::unique_ptr<Graph> MakeOuterGraph(
        const FunctionLibraryDefinition& flib_def, const string& function) {
      Scope scope = Scope::NewRootScope().ExitOnError();
      TF_EXPECT_OK(scope.graph()->AddFunctionLibrary(flib_def.ToProto()));
    
      auto a = ops::Placeholder(scope.WithOpName("A"), DT_INT32);
      auto b = ops::Placeholder(scope.WithOpName("B"), DT_FLOAT);
      auto c = ops::Placeholder(scope.WithOpName("C"), DT_INT32);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 16 18:03:15 UTC 2023
    - 14.7K bytes
    - Viewed (0)
  7. tensorflow/compiler/jit/mark_for_compilation_pass_test.cc

          /*ret_def=*/{{"out", "shape:output:0"}});
    
      func.mutable_signature()->set_is_stateful(true);
      *flib_def.add_function() = std::move(func);
      TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def));
      NodeDef call_node;
      call_node.set_name("fn_call");
      call_node.set_op("Stateful_func");
      Status status;
      Node* call = root.graph()->AddNode(call_node, &status);
      TF_ASSERT_OK(status);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 14 10:11:10 UTC 2024
    - 79.6K bytes
    - Viewed (0)
Back to top