- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for sets (0.15 sec)
-
tensorflow/c/c_api_test.cc
TF_DeleteGraph); TF_SessionOptions* opts = TF_NewSessionOptions(); TF_Session* sess = TF_NewSession(graph.get(), opts, s); TF_DeleteSessionOptions(opts); const string gpu_device_name = GPUDeviceName(sess); TF_DeleteSession(sess, s); CHECK_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); return gpu_device_name; }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3) -
tensorflow/c/experimental/filesystem/plugins/gcs/ram_file_block_cache_test.cc
EXPECT_EQ(offset, 0); num_requests++; memset(buffer, 'x', n); notification.Notify(); // Wait for other thread to issue read. Env::Default()->SleepForMicroseconds(100000); // 0.1 secs TF_SetStatus(status, TF_OK, ""); return n; }; tf_gcs_filesystem::RamFileBlockCache cache(block_size, block_size, 0, fetcher);
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:57 GMT 2021 - 23.2K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/modular_filesystem_test.cc
return status.ok(); } // Tests whether a URI scheme results in a filesystem that is supported. // // As we need these URI schemes to instantiate the test suite when // `testing::InitGoogleTest` gets called, here we just store them to an // internal scheme registry. See `URISchemeRegister` above. static bool GetURIScheme(const std::string& scheme) { tensorflow::SchemeVector()->push_back(scheme); return true;
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/dlpack.cc
TensorReference reference; std::vector<int64_t> shape; std::vector<int64_t> strides; DLManagedTensor tensor; explicit TfDlManagedTensorCtx(const TensorReference& ref) : reference(ref) {} }; // Gets tensor from eager tensor handle. const Tensor* GetTensorFromHandle(TFE_TensorHandle* h, TF_Status* status) { if (h == nullptr) { status->status = tensorflow::errors::InvalidArgument("Invalid handle"); return nullptr;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 12.8K bytes - Viewed (0) -
tensorflow/c/eager/c_api_distributed_test.cc
} void TestFunctionWithPackedInput(const bool remote) { tensorflow::ServerDef server_def = GetServerDef(3); // This server def has the task index set to 0. string serialized = server_def.SerializeAsString(); server_def.set_task_index(1); std::unique_ptr<tensorflow::GrpcServer> worker_server1; ASSERT_TRUE(tensorflow::GrpcServer::Create(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
// A ParallelDevice on its own is not registered with a TFE_Context, and so has // no device name (e.g. for `tf.device`). `NamedParallelDevice` associates a // name with it, which lets us pack its `ParallelTensor`s into TFE_TensorHandles // placed on the parallel device. class NamedParallelDevice { public: NamedParallelDevice(const std::string& name,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.cc
if (prefix == nullptr || strlen(prefix) == 0) prefix = "/job:localhost/replica:0/task:0"; tensorflow::SessionOptions sess_options; (*sess_options.config.mutable_device_count())["CPU"] = num_cpus; status->status = tensorflow::DeviceFactory::AddCpuDevices(sess_options, prefix, &devices); // Remove the device that has the host device name since host device is alreay // in an initialized context.
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 11 23:52:39 GMT 2024 - 35.9K bytes - Viewed (3) -
tensorflow/c/eager/c_api.cc
TFE_ContextSetServerDefWithTimeoutAndRetries( ctx, keep_alive_secs, proto, proto_len, /*init_timeout_in_ms=*/0, /*retries=*/0, status, /*clear_existing_contexts=*/false); } // Set server def with timeout. TF_CAPI_EXPORT extern void TFE_ContextSetServerDefWithTimeout( TFE_Context* ctx, int keep_alive_secs, const void* proto, size_t proto_len, int64_t init_timeout_in_ms, TF_Status* status,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2)