- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for tf_mlir_enable_multiple_local_cpu_devices (4.42 sec)
-
tensorflow/compiler/jit/flags.cc
Flag("tf_mlir_enable_strict_clusters", &enable_mlir_strict_clusters, "Do not allow clusters that have cyclic control dependencies."), Flag("tf_mlir_enable_multiple_local_cpu_devices", &enable_mlir_multiple_local_cpu_devices, "Enable multiple local CPU devices. CPU ops which are outside " "compiled inside the tpu cluster will also be replicated across "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 24.5K bytes - Viewed (0) -
tensorflow/compiler/jit/flags.h
bool tf_mlir_enable_tpu_variable_runtime_reformatting_pass; // TODO(pineapplejuice233): Revisit this flag once the performance impact is verified // with different local CPU devices settings. bool tf_mlir_enable_multiple_local_cpu_devices; }; // Flags for the JitRt pipeline -- see tf_jitrt_pipeline.h for details. struct JitRtFlags { bool always_specialize; bool cost_driven_async_parallel_for;
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/mlir/tensorflow/utils/tpu_rewrite_device_util.cc
ParsedDevice tpu_device, ParsedDevices devices) { tpu_device.type = DEVICE_CPU; bool enable_multiple_local_cpu_devices = tensorflow::GetMlirCommonFlags() ->tf_mlir_enable_multiple_local_cpu_devices; if (!enable_multiple_local_cpu_devices) { tpu_device.id = 0; } if (FindMatchingDevices(devices, tpu_device).empty()) { return absl::InvalidArgumentError(absl::StrCat(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 10 20:10:40 UTC 2024 - 32.8K bytes - Viewed (0)