- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for flib_ (0.05 sec)
-
tensorflow/compiler/mlir/mlir_graph_optimization_pass_test.cc
// We expect the result graph to be exactly the same as the original graph // 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;
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/tensorflow/translate/import_model.cc
explicit NameUniquifier(const FunctionLibraryDefinition& flib) : flib_(flib) {} private: bool IsUnique(llvm::StringRef name) override { return !flib_.Contains(std::string(name)); } std::string GetName(OpOrVal op_or_val) override { DCHECK(false) << "Unimplemented"; return ""; } const FunctionLibraryDefinition& flib_; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util_test.cc
} 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); ConfigProto config = ConfigProto();
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/mark_for_compilation_pass_test_helper.cc
} /*static*/ Status MarkForCompilationPassTestHelper::MarkForCompilation( std::unique_ptr<Graph>* graph, MarkForCompilationPassTestHelper::Options options) { FunctionDefLibrary flib; FunctionLibraryDefinition flib_def((*graph)->op_registry(), flib); return MarkForCompilation(graph, &flib_def, options); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 09 19:51:48 UTC 2023 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_kernel_creator_test.cc
void Init(const std::vector<FunctionDef>& flib) { SessionOptions options; auto* device_count = options.config.mutable_device_count(); device_count->insert({"CPU", 1}); std::vector<std::unique_ptr<Device>> devices; TF_CHECK_OK(DeviceFactory::AddDevices( options, "/job:localhost/replica:0/task:0", &devices)); FunctionDefLibrary proto; for (const auto& fdef : flib) { *(proto.add_function()) = fdef;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 16 01:39:55 UTC 2023 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_registration.cc
return Status(absl::StatusCode::kInvalidArgument, "Invalid graph or client"); } tensorflow::FunctionDefLibrary flib; auto flib_def = std::make_unique<tensorflow::FunctionLibraryDefinition>( tensorflow::OpRegistry::Global(), flib); tensorflow::XlaCompiler::Options options; options.device_type = tensorflow::DeviceType(kArbitraryDeviceName); options.client = client;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_xla_computations_pass_test.cc
} TEST(EncapsulateXlaComputations, BuildXlaLaunchOp) { std::unique_ptr<Graph> body_graph = MakeBodyGraph(); FunctionDefLibrary flib; TF_ASSERT_OK(GraphToFunctionDef(*body_graph, "launch0", flib.add_function())); FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib); std::unique_ptr<Graph> graph = MakeOuterGraph(flib_def, "launch0");
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
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/jit/xla_cluster_util_test.cc
} absl::StatusOr<std::vector<string>> GetNodesRelatedToRefVarsSorted( const Scope& scope, FunctionLibraryDefinition* flib_def = nullptr) { FunctionDefLibrary flib; FunctionLibraryDefinition flib_def_local(OpRegistry::Global(), flib); if (flib_def == nullptr) { flib_def = &flib_def_local; } std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global()));
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/v2/legalize_tf.cc
<< std::get<0>(computation).mlir_module; break; case 1: { auto input = std::get<1>(computation); Graph g(input.flib_def); VLOG(2) << "LegalizeMlirToHlo with FLIB computation input: " << DumpGraphToFile( absl::StrCat("legalize_mlir_hlo_computation_input_", input.function->name()),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 00:40:46 UTC 2024 - 6.8K bytes - Viewed (0)