- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for add_function (0.29 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/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/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) -
tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf_test.cc
} TEST(LegalizeTFTest, RecordsStreamzForNoMlirFallback) { FunctionDef my_func = tensorflow::FunctionDefHelper::Create("empty", {}, {}, {}, {}, {}); tensorflow::FunctionDefLibrary fdef; *(fdef.add_function()) = my_func; tensorflow::FunctionLibraryDefinition flib_def( tensorflow::OpRegistry::Global(), fdef); OpInputList guaranteed_constants; NameAttrList function;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 16.1K bytes - Viewed (0) -
src/cmd/vendor/github.com/google/pprof/internal/symbolizer/symbolizer.go
} symbolizeOneMapping(m, locs, f, addFunction) f.Close() } if missingBinaries { ui.PrintErr("Some binary filenames not available. Symbolization may be incomplete.\n" + "Try setting PPROF_BINARY_PATH to the search path for local binaries.") } return nil } func symbolizeOneMapping(m *profile.Mapping, locs []*profile.Location, obj plugin.ObjFile, addFunction func(*profile.Function) *profile.Function) {
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Fri May 31 19:48:28 UTC 2024 - 10K bytes - Viewed (0)