- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for use_global_compute_stream_ (0.74 sec)
-
tensorflow/compiler/jit/xla_device.cc
use_multiple_streams_(options.use_multiple_streams), shape_determination_fns_(options.shape_determination_fns), allowed_devices_(options.allowed_devices), use_global_compute_stream_(options.use_global_compute_stream) { if (options.shape_determination_fns.empty()) { LOG(ERROR) << "shape_representation_fns must be non-empty."; } VLOG(1) << "Created XLA device " << options.compilation_device_name << " "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 21:05:42 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_tpu_device.cc
options.device_ordinal = i; options.compilation_device_name = DEVICE_TPU_XLA_JIT; options.use_multiple_streams = true; // We set `use_global_compute_stream` to true for TPUs as TPUs can only // have one program running on each core at the same time. options.use_global_compute_stream = true; XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns{ UseNoPreferenceLayoutFn(), &TpuShapeRepresentation};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 20.9K bytes - Viewed (0)