- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for num_devices (0.1 sec)
-
tensorflow/c/eager/c_api_test_util.cc
TF_DeviceList* devices = TFE_ContextListDevices(ctx, status.get()); CHECK_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get()); const int num_devices = TF_DeviceListCount(devices); for (int i = 0; i < num_devices; ++i) { const string dev_type(TF_DeviceListType(devices, i, status.get())); CHECK_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental_test.cc
float* m_float = static_cast<float*>(TF_TensorData(m_data)); TF_DeviceList* devices = TFE_ContextListDevices(ctx, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); int num_devices = TF_DeviceListCount(devices); for (int d = 0; d < num_devices; ++d) { const char* name = TF_DeviceListName(devices, d, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 03 03:14:26 UTC 2023 - 31.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/replicate_to_island.mlir
// CHECK: %{{.*}} = tf_executor.island(%[[CT_0]], %[[CT_1]]) // Tests devices are not remapped if no devices were defined in replicate. // CHECK-LABEL: func @no_devices func.func @no_devices() { tf_executor.graph { %0 = tf_executor.island { tf_device.replicate {n = 2 : i32} { "tf_device.launch"() ({ "tf.opA"() : () -> () tf_device.return
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 15.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/layout_optimization.cc
return signalPassFailure(); // If there is no runtime device information and data format is not explicitly // forced, there is nothing to do. if (devices.NumDevices() == 0 && force_data_format_.empty()) return; func.walk([&](LayoutSensitiveInterface layout_sensitive_interface) { // Get desired op data format. StringRef target_data_format = force_data_format_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.3K bytes - Viewed (0)