- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for DisableXlaCompilation (0.32 sec)
-
tensorflow/compiler/jit/device_compiler_disable_test.cc
for (int i = 0; i < 2; ++i) { args[i].kind = XlaCompiler::Argument::kParameter; args[i].type = DT_INT32; args[i].shape = TensorShape({2, i + 1}); args[i].name = absl::StrCat("arg", i); } DisableXlaCompilation(); xla::LocalClient* client = xla::ClientLibrary::LocalClientOrDie(); DeviceType device_type = DeviceType(DEVICE_CPU_XLA_JIT); const XlaCompiler::CompilationResult* compilation_result;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/jit/flags.h
// be used by a server to ensure that JIT compilation is opt-in. void DisableXlaCompilation(); // Enables XLA compilation. Can be used with `DisableXlaCompilation` to // enable/disable JIT compilation at different stages. void EnableXlaCompilation(); // Returns `false` unless `DisableXlaCompilation` was called. bool FailOnXlaCompilation(); } // namespace tensorflow
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/flags.cc
absl::call_once(flags_init, &AllocateAndParseFlags); AppendMarkForCompilationPassFlagsInternal(flag_list); } static std::atomic<bool> xla_compilation_disabled(false); void DisableXlaCompilation() { xla_compilation_disabled = true; } void EnableXlaCompilation() { xla_compilation_disabled = false; } bool FailOnXlaCompilation() { return xla_compilation_disabled; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 24.5K bytes - Viewed (0)