- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 31 for tf_library (0.31 sec)
-
tensorflow/compiler/aot/BUILD
# tfcompile.bzl correctly handles usage from outside of the package that it is # defined in. # A simple test of tf_library from a text protobuf, to enable benchmark_test. # This test uses an incompleted graph with a node that is not defined. The # compilation works because the undefined node is a feed node. tf_library( name = "test_graph_tfadd", testonly = 1, config = "test_graph_tfadd.config.pbtxt",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 11 16:13:05 UTC 2024 - 11.7K bytes - Viewed (0) -
tensorflow/compiler/aot/tests/BUILD
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 27 18:00:18 UTC 2024 - 17.3K bytes - Viewed (0) -
tensorflow/compiler/aot/tfcompile.bzl
"""Build macro that compiles a TensorFlow graph into a cc_library. To use from your BUILD file, add the following line to load the macro: load("//tensorflow/compiler/aot:tfcompile.bzl", "tf_library") Then call the macro like this: tf_library( name = "test_graph_tfmatmul", config = "test_graph_tfmatmul.config.pbtxt", cpp_class = "MatMulComp", graph = ":test_graph_tfmatmul.pb", ) """ load(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 19:18:08 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/c/c_api_experimental_test.cc
string lib_path = tensorflow::GetDataDependencyFilepath(tensorflow::io::JoinPath( "tensorflow", "c", "experimental", "stream_executor", "test", "test_pluggable_device.so")); TF_Library* lib = TF_LoadPluggableDeviceLibrary(lib_path.c_str(), status); TF_Code code = TF_GetCode(status); string status_msg(TF_Message(status)); TF_DeleteStatus(status); ASSERT_EQ(TF_OK, code) << status_msg;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 17 22:27:52 UTC 2023 - 13.1K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
// for the first time. TF_Library* TF_LoadPluggableDeviceLibrary(const char* library_filename, TF_Status* status) { #if defined(IS_MOBILE_PLATFORM) || defined(IS_SLIM_BUILD) status->status = tensorflow::errors::Unimplemented( "PluggableDevice plugin functionality is not supported on mobile"); return nullptr; #else TF_Library* lib_handle = new TF_Library;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.h
// // On failure, returns nullptr and places an error status in status. TF_CAPI_EXPORT extern TF_Library* TF_LoadPluggableDeviceLibrary( const char* library_filename, TF_Status* status); // Frees the memory associated with the library handle. // Does NOT unload the library. TF_CAPI_EXPORT extern void TF_DeletePluggableDeviceLibraryHandle( TF_Library* lib_handle);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 15.1K bytes - Viewed (0) -
tensorflow/compiler/aot/compile.cc
TF_ASSIGN_OR_RETURN(std::unique_ptr<xla::HloSnapshot> module, computation.Snapshot()); // Serialize the HloSnapshot deterministically so that all the outputs of a // tf_library genrule are deterministic. const size_t size = module->ByteSizeLong(); auto serialized = std::make_unique<char[]>(size); TF_RET_CHECK( SerializeToBufferDeterministic(*module, serialized.get(), size));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 08:28:57 UTC 2024 - 11.9K bytes - Viewed (0) -
tensorflow/compiler/aot/tests/tfcompile_test.cc
#include "xla/test.h" #include "tensorflow/core/platform/regexp.h" #include "tensorflow/core/platform/test.h" // The header files for the tests using mlir_bridge have the _mlir_bridge suffix // inherited from the tf_library target names. #if defined(ENABLE_MLIR_BRIDGE_TEST) #include "tensorflow/compiler/aot/tests/test_graph_tfadd_mlir_bridge.h" #include "tensorflow/compiler/aot/tests/test_graph_tfadd_with_ckpt_mlir_bridge.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 26.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/BUILD
load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library", "td_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") load("//tensorflow:tensorflow.default.bzl", "filegroup", "get_compatible_with_portable") load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") # copybara:uncomment_end(google-only) package( # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"], default_visibility = ["//visibility:public"],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 35.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/BUILD
"@llvm-project//mlir:TransformUtils", ], ) cc_library( name = "constants", hdrs = ["constants.h"], ) cc_library( name = "test_opkernels", testonly = True, srcs = ["test_opkernels.cc"], deps = ["//tensorflow/core:framework"], alwayslink = True, ) cc_library( name = "backend_compiler", srcs = ["backend_compiler.cc"],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 19:04:21 UTC 2024 - 23.6K bytes - Viewed (0)