- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 25 for define (0.16 sec)
-
tensorflow/c/eager/gradients.cc
namespace internal { Status Reset(AbstractOperation* op_, const char* op, const char* raw_device_name, ForwardOperation* forward_op_) { forward_op_->op_name = op; forward_op_->attrs.Reset(op); return op_->Reset(op, raw_device_name); } Status AddInput(AbstractOperation* op_, AbstractTensorHandle* input, ForwardOperation* forward_op_) {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 19.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_distributed_test.cc
const string composite_device_name = "/job:localhost/replica:0/task:0/device:COMPOSITE:0"; EXPECT_EQ(TFE_TensorHandleDeviceName(packed_handle, status), composite_device_name); EXPECT_EQ(TFE_TensorHandleBackingDeviceName(packed_handle, status), composite_device_name); ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/gcs/expiring_lru_cache_test.cc
// re-read the same value, expect another lookup cache1.LookupOrCompute("a", &value, compute_func, &status); TF_EXPECT_OK(status.status); EXPECT_EQ(value, 1); EXPECT_EQ(num_compute_calls, 2); // Define a new cache with max_age > 0 and verify correct behavior. tf_gcs_filesystem::ExpiringLRUCache<int> cache2(2, 4); num_compute_calls = 0; value = -1; // Read our first value
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Jul 09 19:31:22 GMT 2020 - 7.1K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
const char* const* underlying_devices, int num_underlying_devices, TFE_CustomDevice* device, void** device_info) { device->copy_tensor_to_device = &CopyToParallelDevice; device->copy_tensor_from_device = &CopyTensorFromParallelDevice; device->delete_device = &DeleteParallelDevice; device->execute = &ParallelDeviceExecute;
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
absl::optional<int64_t> step_id) const { for (int device_index = 0; device_index < underlying_devices_.size(); ++device_index) { DeviceThread* device_thread = device_threads_[device_index].get(); std::vector<TFE_TensorHandle*> device_inputs; device_inputs.reserve(inputs.size()); for (int input_index = 0; input_index < inputs.size(); ++input_index) {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
const char* device_name = "/job:localhost/replica:0/task:0/device:CUSTOM:0"; const char* first_device_name = "/job:localhost/replica:0/task:0/device:CPU:0"; const char* second_device_name = "/job:localhost/replica:0/task:0/device:CPU:1"; std::array<const char*, 2> underlying_devices{first_device_name, second_device_name};
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 29.3K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
// Create and modify a variable placed on a parallel device which composes // `first_device` and `second_device`. void BasicTestsForTwoDevices(TFE_Context* context, const char* first_device, const char* second_device) { // Register the custom device std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status( TF_NewStatus(), TF_DeleteStatus);
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Jun 15 15:44:44 GMT 2021 - 12.5K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_testutil.cc
TFE_CustomDevice custom_device; custom_device.copy_tensor_to_device = &CopyToLoggingDevice; custom_device.copy_tensor_from_device = &CopyTensorFromLoggingDevice; custom_device.delete_device = &DeleteLoggingDevice; custom_device.execute = &LoggingDeviceExecute; LoggingDevice* device = new LoggingDevice; device->arrived_flag = arrived_flag; device->executed_flag = executed_flag;
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Wed Mar 03 20:47:31 GMT 2021 - 8.3K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
CHECK_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); LOG(INFO) << "Device " << i << " has name " << device_name << ", type " << device_type; if (string(device_type) == DEVICE_GPU) { return device_name; } } // No GPU device found. return ""; } string GPUDeviceName() { std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status(
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3) -
tensorflow/c/c_api_experimental_test.cc
ASSERT_EQ(TF_OK, code) << status_msg; TF_DeletePluggableDeviceLibraryHandle(lib); #endif // !defined(TENSORFLOW_NO_SHARED_OBJECTS) #endif // !defined(PLATFORM_WINDOWS) } TEST(CAPI_EXPERIMENTAL, LibraryNextPluggableDeviceLoadFunctions) { // TODO(penpornk): Enable this test on Windows. #if !defined(PLATFORM_WINDOWS) #if !defined(TENSORFLOW_NO_SHARED_OBJECTS) // Load the library. TF_Status* status = TF_NewStatus();
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Jan 17 22:27:52 GMT 2023 - 13.1K bytes - Viewed (1)