- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for flib_ (0.04 sec)
-
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/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) -
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_cl.cc
llvm::cl::desc("Set original TF function name on importi."), llvm::cl::init(false)); // Export options. // NOLINTNEXTLINE opt<bool> export_entry_func_to_flib( "tf-export-entry-func-to-flib", llvm::cl::desc( "Export entry function to function library instead of graph"), llvm::cl::init(false)); // NOLINTNEXTLINE opt<bool> export_original_tf_func_name( "tf-export-original-func-name",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 10 20:59:50 UTC 2023 - 5.5K bytes - Viewed (0)