Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for ShouldCompileCluster (0.18 sec)

  1. 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)
  2. 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)
Back to top