Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 21 for add_function (0.18 sec)

  1. tensorflow/compiler/jit/rearrange_function_argument_pass_test.cc

        auto ret1 = ops::_Retval(s.WithOpName("ret1"), arg0, 1);
        std::unique_ptr<Graph> g(new Graph(OpRegistry::Global()));
        TF_CHECK_OK(s.ToGraph(g.get()));
        FunctionDef *xla_fdef = fdl.add_function();
        TF_CHECK_OK(GraphToFunctionDef(*g, "f1", xla_fdef));
      }
      {
        // Function for While's "body".
        // "arg0" (T=DT_RESOURCE), "arg1" (T=DT_BOOL)
        // "ret0" = "arg0"
        // "ret1" = "arg1"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 09 11:36:41 UTC 2024
    - 10.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util_test.cc

          {{{"out"}, "Identity", {"in"}, {{"T", DataType::DT_RESOURCE}}}});
    }
    
    TEST(IsSupportedByNonReplicatedBridge, NonReplicatedGraph) {
      const FunctionDef& fd = PassThroughResource();
      FunctionDefLibrary flib;
      *flib.add_function() = fd;
      FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib);
      Graph graph(flib_def);
      graph.SetConstructionContext(ConstructionContext::kEagerRuntime);
      tensorflow::set_tf2_execution(true);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 17 19:51:50 UTC 2024
    - 10.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc

        PartialTensorShape shape({2});
        node_name_image["identity1"]->AddAttr(
            kXlaInferredShapesAttrName, std::vector<PartialTensorShape>{shape});
    
        FunctionDef *xla_fdef = fdl.add_function();
        TF_CHECK_OK(GraphToFunctionDef(*g, "cluster", xla_fdef));
      }
      FunctionLibraryDefinition fld(OpRegistry::Global(), fdl);
    
      protobuf::Map<string, tensorflow::AttrValue> attrs;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 41K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc

      GraphDef graphdef_expected;
    
      *library_expected.add_function() = test::function::XTimesTwo();
      *library_expected.add_function() = FunctionDefHelper::Create(
          "F1", {"a_0_arg:float"}, {"c_0_retval:float"}, {},
          {
              {{"C"}, "UnaryTest", {"a_0_arg"}},
          },
          {{"c_0_retval", "C:o:0"}});
      *library_expected.add_function() = FunctionDefHelper::Create(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 113.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/force_xla_constants_on_host_pass_test.cc

                                    {{"c", "t0:y:0"}});
    
      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);
    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/tests/device_compiler_test_helper.cc

           {{"k"}, "Add", {"j", "h"}, {{"T", DT_FLOAT}}},
           {{"l"}, "Add", {"k", "h"}, {{"T", DT_FLOAT}}},
           {{"m"}, "Identity", {"l"}, {{"T", DT_FLOAT}}}});
    
      GraphDef graph;
      *graph.mutable_library()->add_function() = make_test_fn;
      *graph.add_node() = MakeNode("a", "Placeholder", {},
                                   {{"dtype", DT_FLOAT}, {"shape", input_shape}});
      *graph.add_node() = MakeNode("b", "Placeholder", {},
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 09 08:24:16 UTC 2024
    - 6.2K bytes
    - Viewed (0)
  7. tensorflow/compiler/jit/build_xla_ops_pass_test.cc

          /*function_name=*/name, /*in_def=*/{}, /*out_def=*/{"out: float"},
          /*attr_def*/
          {}, /*node_def=*/{FunctionDefHelper::Const("one", 1.0f)},
          /*ret_def=*/{{"out", "out:output:0"}});
      *fdef_lib.add_function() = std::move(func);
      return fdef_lib;
    }
    
    TEST_F(BuildXlaOpsTest, ControlDepsPreserved) {
      const char* kXlaDeviceName = "/job:worker/replica:0/task:0/device:XLA_CPU:0";
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 12.2K bytes
    - Viewed (0)
  8. tensorflow/compiler/jit/xla_cluster_util_test.cc

          {{{"var"},
            "VariableV2",
            {},
            {{"dtype", DT_FLOAT}, {"shape", TensorShape({})}}},
           {{"r"}, "Identity", {"var"}, {{"T", DT_FLOAT}}}});
      *fdef_lib->add_function() = make_ref_float;
    }
    
    void AddRegularFunctionFunctionDef(FunctionDefLibrary* fdef_lib) {
      Tensor seven(DT_FLOAT, {});
      seven.scalar<float>()() = 7;
      FunctionDef make_regular_float = FunctionDefHelper::Define(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 10.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/jit/mark_for_compilation_pass_test.cc

      noinline.mutable_signature()->set_name("NoInlineFn");
      AddAttr("_noinline", static_cast<bool>(true), noinline.mutable_attr());
    
      FunctionDefLibrary flib;
      *flib.add_function() = compilable;
      *flib.add_function() = uncompilable;
      *flib.add_function() = noinline;
      FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib);
    
      std::unique_ptr<Graph> graph(new Graph(&flib_def));
      {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 14 10:11:10 UTC 2024
    - 79.6K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph_test.cc

      FunctionDef empty_function =
          tensorflow::FunctionDefHelper::Create("empty", {}, {}, {}, {}, {});
    
      tensorflow::FunctionDefLibrary fdef;
      *(fdef.add_function()) = empty_function;
      tensorflow::FunctionLibraryDefinition flib_def(
          tensorflow::OpRegistry::Global(), fdef);
    
      OpInputList guaranteed_constants;
      NameAttrList function;
      function.set_name("empty");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 08:08:57 UTC 2024
    - 11.7K bytes
    - Viewed (0)
Back to top