- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 16 for flib_ (0.07 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/jit/compilability_check_util_test.cc
FunctionDefLibrary flib; *flib.add_function() = FunctionDefHelper::Define( /*Function*/ kCompilableFunctionName, /*Inputs*/ {"n_a:float", "n_b:float"}, /*Outputs*/ {"n_c:float"}, /*Attribute*/ {}, // Node info {{{kCompilableFunctionNodeName}, "Add", {"n_a", "n_b"}, {{"T", DT_FLOAT}}}}); *flib.add_function() = FunctionDefHelper::Define(
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/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/mlir/tensorflow/tests/mlir2graphdef/export_main_to_flib.mlir
// RUN: tf-mlir-translate -mlir-to-graphdef -tf-export-entry-func-to-flib %s -o - 2>&1 | FileCheck %s module attributes {tf.versions = {bad_consumers = [], min_consumer = 12 : i32, producer = 458 : i32}} { func.func @main() attributes {tf.entry_function = {inputs = "", outputs = ""}} { tf_executor.graph {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 699 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)