- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for ShouldCompileCluster (0.29 sec)
-
tensorflow/compiler/jit/device_compiler_test.cc
XlaCompiler::Options options = GetDefaultXlaOptions(); NameAttrList fn; fn.set_name("foo"); // Using a mock here since it's difficult to have ShouldCompileCluster() // return false. EXPECT_CALL(*mock_profiler_, ShouldCompileCluster(_, DeviceCompileMode::kLazy, 1)) .WillOnce(Return(false)); TF_EXPECT_OK(xla_device_compiler_->CompileIfNeeded(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 19.8K bytes - Viewed (0) -
tensorflow/compiler/jit/device_compiler.h
return errors::Internal("XLA compilation disabled"); } if (state == DeviceCompileState::kUncompiled) { XLA_SCOPED_LOGGING_TIMER("Compilation of XLA executable"); if (!profiler->ShouldCompileCluster(function, compile_mode, current_request_count)) { VLOG(2) << "Not compiling for signature: " << human_signature; return absl::OkStatus();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 22.1K bytes - Viewed (0)