OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2026-01-15 12:15:41 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Automated Code Change
Some checks failed
ARM CD / build (3.10) (push) Has been cancelled
Details
ARM CD / build (3.11) (push) Has been cancelled
Details
ARM CD / build (3.12) (push) Has been cancelled
Details
ARM CD / build (3.9) (push) Has been cancelled
Details
ARM CI / build (3.10) (push) Has been cancelled
Details
Creates a GitHub Issue when a PR Rolled back via Commit to Master / create-issue-on-pr-rollback (push) Has been cancelled
Details
Scorecards supply-chain security / Scorecards analysis (push) Has been cancelled
Details
Close inactive issues / close-issues (push) Has been cancelled
Details
ARM CI Extended C++ / build (3.10) (push) Has been cancelled
Details
Set nightly branch to master HEAD / master-to-nightly (push) Has been cancelled
Details
ARM CI Extended / build (3.10) (push) Has been cancelled
Details
ARM CI Extended / build (3.11) (push) Has been cancelled
Details
ARM CI Extended / build (3.12) (push) Has been cancelled
Details
ARM CI Extended / build (3.9) (push) Has been cancelled
Details
OSV-Scanner Scheduled Scan / scan-scheduled (push) Has been cancelled
Details

Browse Source 
PiperOrigin-RevId: 851260918
This commit is contained in:
A. Unique TensorFlower
2026-01-02 02:38:08 -08:00
committed by TensorFlower Gardener
parent 314ef8094a
commit c88e652871
4 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
third_party/xla/xla/service/compile_only_service.cc vendored
View File

@@ -61,7 +61,7 @@ CompileOnlyService::CompileOnlyService(const ServiceOptions& options,
: Service(options, /*execute_backend=*/nullptr),
compiler_(std::move(compiler)) {}
absl::StatusOr<std::vector<std::unique_ptr<AotCompilationResult>>>
absl::StatusOr<std::vector<std::unique_ptr<CompiledModule>>>
CompileOnlyService::CompileAheadOfTime(
const AotXlaComputationInstance& computation,
const AotCompilationOptions& options,

2
third_party/xla/xla/service/compile_only_service.h vendored
View File

@@ -48,7 +48,7 @@ class CompileOnlyService : public Service {
// Compiles a list of xla computations for ahead-of-time execution. This is
// intended for use in static compilation. See
// |CompileOnlyClient::CompileAheadOfTime| for additional details.
absl::StatusOr<std::vector<std::unique_ptr<AotCompilationResult>>>
absl::StatusOr<std::vector<std::unique_ptr<CompiledModule>>>
CompileAheadOfTime(const AotXlaComputationInstance& computation,
const AotCompilationOptions& options,
std::unique_ptr<AotCompilationMetadata>* metadata);

2
third_party/xla/xla/service/compiler.cc vendored
View File

@@ -50,7 +50,7 @@ std::unique_ptr<tsl::protobuf::Message> Compiler::ComputeDefaultBackendConfig(
}
// Define a default version where metadata is not used.
absl::StatusOr<std::vector<std::unique_ptr<AotCompilationResult>>>
absl::StatusOr<std::vector<std::unique_ptr<CompiledModule>>>
Compiler::CompileAheadOfTime(
std::unique_ptr<HloModule> hlo_module, const AotCompilationOptions& options,
std::unique_ptr<AotCompilationMetadata>* metadata) {

10
third_party/xla/xla/service/compiler.h vendored
View File

@@ -211,7 +211,7 @@ class Compiler {
// Returns a (deserialized) AotCompilationResult from a serialized
// AotCompilationResult.
virtual absl::StatusOr<std::unique_ptr<AotCompilationResult>>
virtual absl::StatusOr<std::unique_ptr<CompiledModule>>
LoadAotCompilationResult(const std::string& serialized_aot_result) {
return Unimplemented("LoadAotCompilationResult unimplemented.");
}
@@ -255,13 +255,13 @@ class Compiler {
// Compiles the HLO module for ahead-of-time execution. This is intended for
// use in static compilation.
virtual absl::StatusOr<std::vector<std::unique_ptr<AotCompilationResult>>>
virtual absl::StatusOr<std::vector<std::unique_ptr<CompiledModule>>>
CompileAheadOfTime(std::unique_ptr<HloModule> module,
const AotCompilationOptions& options) = 0;
// Similar to CompileAheadOfTime above but AotCompilationMetadata
// has an argument that can be populated during compilation.
virtual absl::StatusOr<std::vector<std::unique_ptr<AotCompilationResult>>>
virtual absl::StatusOr<std::vector<std::unique_ptr<CompiledModule>>>
CompileAheadOfTime(std::unique_ptr<HloModule> module,
const AotCompilationOptions& options,
std::unique_ptr<AotCompilationMetadata>* metadata);
@@ -303,7 +303,7 @@ class Compiler {
}
// Returns an AotCompilationResult of the executable for serialization.
virtual absl::StatusOr<std::unique_ptr<AotCompilationResult>> Export(
virtual absl::StatusOr<std::unique_ptr<CompiledModule>> Export(
Executable* executable) {
return Unimplemented("Export unimplemented");
}
@@ -321,7 +321,7 @@ class Compiler {
// Creates an `Executable` based on the given `aot_result`.
virtual absl::StatusOr<std::unique_ptr<Executable>>
LoadExecutableFromAotResult(const AotCompilationResult& aot_result,
LoadExecutableFromAotResult(const CompiledModule& aot_result,
const se::StreamExecutor& stream_exec) {
return Unimplemented("LoadExecutableFromAotResult unimplemented");
}