- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 15 for add_function (0.14 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
tensorflow/compiler/mlir/mlir_graph_optimization_pass_test.cc
// so we define the `graph_` by the following `flib` in this test point // instead of the way we do in the Init method. FunctionDefLibrary flib; *flib.add_function() = XTimesTwo(); FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib); graph_ = std::make_unique<Graph>(flib_def); GraphDef original_graph_def; graph_->ToGraphDef(&original_graph_def);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 08:25:30 UTC 2024 - 16.1K bytes - Viewed (0)