- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for AllowForDeviceInXlaLaunch (0.49 sec)
-
tensorflow/compiler/jit/xla_compile_util_test.cc
// Flag is turned on, some device is allowlisted, but the requested one isn't. rollout_config.AllowForDeviceInXlaLaunch(DeviceType(DEVICE_GPU)); EXPECT_FALSE(UsePjRtForSingleDeviceCompilation(DeviceType(DEVICE_CPU))); // Flag is turned on and the requested device is allowlisted. rollout_config.AllowForDeviceInXlaLaunch(DeviceType(DEVICE_CPU)); EXPECT_TRUE(UsePjRtForSingleDeviceCompilation(DeviceType(DEVICE_CPU)));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 16 21:48:05 UTC 2023 - 6K bytes - Viewed (0) -
tensorflow/compiler/jit/device_context_test.cc
}(); class DeviceContextTest : public ::testing::Test { public: void SetDevice(const string& device_type) { auto& rollout_config = GetXlaOpsCommonFlags()->tf_xla_use_device_api; rollout_config.AllowForDeviceInXlaLaunch(DeviceType(device_type)); rollout_config.AllowForDeviceInXlaCompileOnDemand(DeviceType(device_type)); auto device_factory = DeviceFactory::GetFactory(device_type); SessionOptions options;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 3.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_gpu_test.cc
GetXlaDeviceFlags()->tf_xla_enable_xla_devices = true; // Add and setup the GPU device. auto device_type = DeviceType(DEVICE_GPU); auto jit_device_type = DeviceType(DEVICE_GPU); rollout_config.AllowForDeviceInXlaLaunch(device_type); rollout_config.AllowForDeviceInXlaCompileOnDemand(device_type); auto device = DeviceFactory::NewDevice(device_type.type_string(), SessionOptions(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 10K bytes - Viewed (0) -
tensorflow/compiler/jit/flags.h
// to the `device_type` being allowed in order to use the Device API for // single device compilation and execution in the XlaLaunch op. void AllowForDeviceInXlaLaunch(const DeviceType& device_type) { xla_launch_allowed_devices_.insert(device_type.type_string()); } bool IsEnabledInXlaLaunchForDevice(const DeviceType& device_type) const {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 14.5K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_test.cc
// for XLA devices. GetXlaDeviceFlags()->tf_xla_enable_xla_devices = true; // Add and setup the XLA_CPU device. auto device_type = DeviceType(DEVICE_XLA_CPU); rollout_config.AllowForDeviceInXlaLaunch(device_type); rollout_config.AllowForDeviceInXlaCompileOnDemand(device_type); auto jit_device_type = DeviceType(DEVICE_CPU_XLA_JIT); auto device =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 28.8K bytes - Viewed (0)