Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for PjRtDeviceCompilerClient (0.23 sec)

  1. tensorflow/compiler/jit/pjrt_device_compiler_client.h

      void WaitForProgramsToFinish() override;
    
      xla::PjRtClient* client() const override { return client_; }
    
     private:
      xla::PjRtClient* const client_;
    
      PjRtDeviceCompilerClient(const PjRtDeviceCompilerClient&) = delete;
      void operator=(const PjRtDeviceCompilerClient&) = delete;
    };
    
    // Generates CompileOptions for PJRT compilation.
    xla::CompileOptions GetPjRtCompileOptions(
        const XlaCompiler::Options& options,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 3.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/pjrt_device_compiler_client.cc

      return std::move(executable);
    }
    
    absl::StatusOr<std::string> PjRtDeviceCompilerClient::SerializeExecutable(
        const xla::PjRtLoadedExecutable& executable) {
      VLOG(1) << "Serializing xla::PjRtLoadedExecutable to string.";
      return executable.SerializeExecutable();
    }
    
    absl::StatusOr<std::string> PjRtDeviceCompilerClient::BuildSerializedExecutable(
        const XlaCompiler::Options& options,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 3.6K bytes
    - Viewed (0)
Back to top