- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 16 for containers (0.27 sec)
-
tensorflow/c/c_api.cc
static void TF_Reset_Helper(const TF_SessionOptions* opt, const char** containers, int ncontainers, TF_Status* status) { std::vector<string> container_names(ncontainers); for (int i = 0; i < ncontainers; ++i) { container_names[i] = containers[i]; } status->status = Reset(opt->options, container_names); } extern "C" {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.cc
limitations under the License. ==============================================================================*/ #include "tensorflow/c/eager/c_api_unified_experimental.h" #include <vector> #include "absl/container/flat_hash_map.h" #include "absl/strings/str_cat.h" #include "tensorflow/c/eager/c_api_unified_experimental_internal.h" #include "tensorflow/c/tf_datatype.h" #include "tensorflow/c/tf_status.h"
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_test.cc
ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT); TFE_OpSetAttrShape(op.get(), "shape", {}, 0, status.get()); TFE_OpSetAttrString(op.get(), "container", "", 0); TFE_OpSetAttrString(op.get(), "shared_name", "", 0); TFE_OpSetDevice(op.get(), name, status.get()); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 27 23:39:24 GMT 2020 - 18.4K bytes - Viewed (0) -
tensorflow/c/checkpoint_reader.cc
} } bool CheckpointReader::HasTensor(const string& name) const { if (reader_ != nullptr) { return reader_->HasTensor(name, nullptr, nullptr); } return v2_reader_->Contains(name); } const TensorSliceReader::VarToShapeMap& CheckpointReader::GetVariableToShapeMap() const { CHECK(var_to_shape_map_); return *var_to_shape_map_; } const TensorSliceReader::VarToDataTypeMap&
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Aug 25 21:29:12 GMT 2023 - 5.5K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/c_api.cc
auto cc_status = cc_ctx->resource_manager()->Create<tensorflow::PluginResource>( container_name, plugin_resource_name, cc_resource_ptr); status->status = cc_status; } void TF_LookupOrCreatePluginResource( TF_OpKernelContext* ctx, const char* container_name, const char* plugin_resource_name, void** result_plugin_resource, void* (*create_func)(void*), void* create_func_args,
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jan 09 00:52:04 GMT 2024 - 13.9K bytes - Viewed (1) -
tensorflow/c/eager/c_api_experimental_test.cc
/*isolate_session_state=*/false, init_timeout_in_ms); // Remote device on `worker2`. const char remote_device[] = "/job:localhost/replica:0/task:2/device:CPU:0"; // `ctx_0`, `ctx_1` contains `remote_device`. { const std::vector<std::string>& device_names = ListDeviceNames(ctx_0); ASSERT_TRUE(std::find(device_names.begin(), device_names.end(),
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 03:14:26 GMT 2023 - 31.5K bytes - Viewed (1) -
tensorflow/c/eager/c_api_test.cc
/*init_timeout_in_ms=*/0); // Remote device on `worker1`. const char remote_device[] = "/job:localhost/replica:0/task:1/device:CPU:0"; // `ctx_0`, `ctx_1`, `ctx_2` contains `remote_device`. { const std::vector<std::string>& device_names = ListDeviceNames(ctx_0); ASSERT_TRUE(std::find(device_names.begin(), device_names.end(),
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1) -
tensorflow/c/experimental/filesystem/modular_filesystem_test.cc
public: // Initializes `root_dir_` to a unique value made of `::testing::TempDir()`, a // static random value unique for all the tests in one invocation, and the // current test name. // // Since the test name contains `/` (due to parameters), this function // replaces `/` with `_`. // // We trade in one extra initialization for readability. ModularFileSystemTest() {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri May 27 20:25:58 GMT 2022 - 71K bytes - Viewed (0) -
tensorflow/c/eager/gradients_test.cc
limitations under the License. ==============================================================================*/ #include "tensorflow/c/eager/gradients.h" #include <memory> #include "absl/container/flat_hash_set.h" #include "absl/types/span.h" #include "tensorflow/c/eager/abstract_context.h" #include "tensorflow/c/eager/abstract_tensor_handle.h" #include "tensorflow/c/eager/c_api_experimental.h"
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
if (TF_GetCode(status) != TF_OK) return nullptr; TFE_OpSetAttrType(op.get(), "dtype", type); TFE_OpSetAttrShape(op.get(), "shape", dims, num_dims, status); TFE_OpSetAttrString(op.get(), "container", "", 0); // Use the special GUID for no buffer sharing // // TODO(allenl): Should we provide a better API for this? AFAIK this is the // only reasonable way to make variables with no aliasing using the eager C
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Jun 15 15:44:44 GMT 2021 - 12.5K bytes - Viewed (0)