- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for AddFunctionLibrary (0.18 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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)