1
0
Fork 0
mirror of https://github.com/tensorflow/haskell.git synced 2024-11-22 19:09:43 +01:00

Update to TensorFlow 2.12 (#292)

* Fix compilation with tensorflow 2.12 and higher

* Update tensorflow submodule to 2.12.0

* Update Dockerfiles to 2.12.0
This commit is contained in:
Jeroen Bransen 2023-08-15 05:36:35 +02:00 committed by GitHub
parent bfd8de5582
commit 925c2e9515
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
8 changed files with 204 additions and 54 deletions

View file

@ -72,7 +72,7 @@ working, the following commands will compile and run the tests.
```
git clone --recursive https://github.com/tensorflow/haskell.git tensorflow-haskell
cd tensorflow-haskell
docker build -t tensorflow/haskell:2.10.1 docker
docker build -t tensorflow/haskell:2.12.0 docker
# TODO: move the setup step to the docker script.
stack --docker setup
stack --docker test
@ -90,7 +90,7 @@ stack --docker build --exec Main
If you want to use GPU you can do:
```
IMAGE_NAME=tensorflow/haskell:2.10.1-gpu
IMAGE_NAME=tensorflow/haskell:2.12.0-gpu
docker build -t $IMAGE_NAME docker/gpu
# TODO: move the setup step to the docker script.
stack --docker --docker-image=$IMAGE_NAME setup

View file

@ -3,7 +3,7 @@
# stack to be installed on the host. This comes at the expense of
# flexibility.
FROM tensorflow/tensorflow:2.10.1
FROM tensorflow/tensorflow:2.12.0
LABEL maintainer="TensorFlow authors <tensorflow-haskell@googlegroups.com>"
# The build context directory is the top of the tensorflow-haskell
@ -36,8 +36,8 @@ RUN \
curl -O -L https://github.com/google/protobuf/releases/download/v3.13.0/protoc-3.13.0-linux-x86_64.zip && \
unzip -d /usr/local protoc-3.13.0-linux-x86_64.zip bin/protoc && \
chmod 755 /usr/local/bin/protoc && \
curl -O https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.10.1.tar.gz && \
tar zxf libtensorflow-cpu-linux-x86_64-2.10.1.tar.gz -C /usr/local && \
curl -O https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.12.0.tar.gz && \
tar zxf libtensorflow-cpu-linux-x86_64-2.12.0.tar.gz -C /usr/local && \
ldconfig && \
stack setup && \
stack test --only-dependencies

View file

@ -1,6 +1,6 @@
# Prepare the image with:
# docker build -t tensorflow/haskell:2.10.1 docker
FROM tensorflow/tensorflow:2.10.1
# docker build -t tensorflow/haskell:2.12.0 docker
FROM tensorflow/tensorflow:2.12.0
LABEL maintainer="TensorFlow authors <tensorflow-haskell@googlegroups.com>"
RUN apt-get update
@ -31,8 +31,8 @@ RUN \
curl -O -L https://github.com/google/protobuf/releases/download/v3.13.0/protoc-3.13.0-linux-x86_64.zip && \
unzip -d /usr/local protoc-3.13.0-linux-x86_64.zip bin/protoc && \
chmod 755 /usr/local/bin/protoc && \
curl -O https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.10.1.tar.gz && \
tar zxf libtensorflow-cpu-linux-x86_64-2.10.1.tar.gz -C /usr/local && \
curl -O https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.12.0.tar.gz && \
tar zxf libtensorflow-cpu-linux-x86_64-2.12.0.tar.gz -C /usr/local && \
ldconfig
ENV LANG en_US.UTF-8

View file

@ -1,6 +1,6 @@
# Prepare the image with:
# docker build -t tensorflow/haskell:2.10.1-gpu docker/gpu
FROM tensorflow/tensorflow:2.10.1-gpu
# docker build -t tensorflow/haskell:2.12.0-gpu docker/gpu
FROM tensorflow/tensorflow:2.12.0-gpu
LABEL maintainer="TensorFlow authors <tensorflow-haskell@googlegroups.com>"
RUN apt-get update
@ -31,8 +31,8 @@ RUN \
curl -O -L https://github.com/google/protobuf/releases/download/v3.13.0/protoc-3.13.0-linux-x86_64.zip && \
unzip -d /usr/local protoc-3.13.0-linux-x86_64.zip bin/protoc && \
chmod 755 /usr/local/bin/protoc && \
curl -O https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-linux-x86_64-2.10.1.tar.gz && \
tar zxf libtensorflow-gpu-linux-x86_64-2.10.1.tar.gz -C /usr/local && \
curl -O https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-linux-x86_64-2.12.0.tar.gz && \
tar zxf libtensorflow-gpu-linux-x86_64-2.12.0.tar.gz -C /usr/local && \
ldconfig
ENV LANG en_US.UTF-8

View file

@ -29,7 +29,7 @@ extra-include-dirs:
docker:
enable: false
image: tensorflow/haskell:2.10.1
image: tensorflow/haskell:2.12.0
nix:
enable: false

View file

@ -13,6 +13,7 @@ build-type: Custom
cabal-version: 2.0
extra-source-files: third_party/tensorflow/tensorflow/compiler/jit/*.proto
, third_party/tensorflow/tensorflow/compiler/mlir/lite/quantization/*.proto
, third_party/tensorflow/tensorflow/compiler/mlir/quantization/stablehlo/*.proto
, third_party/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/*.proto
, third_party/tensorflow/tensorflow/compiler/mlir/tfrt/analysis/*.proto
, third_party/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/*.proto
@ -20,18 +21,23 @@ extra-source-files: third_party/tensorflow/tensorflow/compiler/jit/*.proto
, third_party/tensorflow/tensorflow/compiler/tf2xla/*.proto
, third_party/tensorflow/tensorflow/compiler/tf2xla/kernels/*.proto
, third_party/tensorflow/tensorflow/compiler/xla/*.proto
, third_party/tensorflow/tensorflow/compiler/xla/mlir/tools/mlir_replay/public/*.proto
, third_party/tensorflow/tensorflow/compiler/xla/pjrt/*.proto
, third_party/tensorflow/tensorflow/compiler/xla/pjrt/distributed/*.proto
, third_party/tensorflow/tensorflow/compiler/xla/python/tpu_driver/*.proto
, third_party/tensorflow/tensorflow/compiler/xla/rpc/*.proto
, third_party/tensorflow/tensorflow/compiler/xla/runtime/runner/*.proto
, third_party/tensorflow/tensorflow/compiler/xla/service/*.proto
, third_party/tensorflow/tensorflow/compiler/xla/service/cpu/*.proto
, third_party/tensorflow/tensorflow/compiler/xla/service/gpu/*.proto
, third_party/tensorflow/tensorflow/compiler/xla/stream_executor/*.proto
, third_party/tensorflow/tensorflow/compiler/xla/tools/*.proto
, third_party/tensorflow/tensorflow/compiler/xrt/*.proto
, third_party/tensorflow/tensorflow/core/data/*.proto
, third_party/tensorflow/tensorflow/core/data/service/*.proto
, third_party/tensorflow/tensorflow/core/debug/*.proto
, third_party/tensorflow/tensorflow/core/example/*.proto
, third_party/tensorflow/tensorflow/core/framework/*.proto
, third_party/tensorflow/tensorflow/core/function/polymorphism/*.proto
, third_party/tensorflow/tensorflow/core/function/trace_type/*.proto
, third_party/tensorflow/tensorflow/core/grappler/costs/*.proto
, third_party/tensorflow/tensorflow/core/grappler/optimizers/inference/*.proto
@ -44,12 +50,14 @@ extra-source-files: third_party/tensorflow/tensorflow/compiler/jit/*.proto
, third_party/tensorflow/tensorflow/core/tpu/kernels/*.proto
, third_party/tensorflow/tensorflow/core/util/*.proto
, third_party/tensorflow/tensorflow/core/util/autotune_maps/*.proto
, third_party/tensorflow/tensorflow/core/util/quantization/*.proto
, third_party/tensorflow/tensorflow/distribute/experimental/rpc/proto/*.proto
, third_party/tensorflow/tensorflow/dtensor/proto/*.proto
, third_party/tensorflow/tensorflow/lite/experimental/acceleration/configuration/*.proto
, third_party/tensorflow/tensorflow/lite/python/metrics/*.proto
, third_party/tensorflow/tensorflow/lite/toco/*.proto
, third_party/tensorflow/tensorflow/lite/toco/logging/*.proto
, third_party/tensorflow/tensorflow/lite/tools/delegates/compatibility/protos/*.proto
, third_party/tensorflow/tensorflow/lite/tools/evaluation/proto/*.proto
, third_party/tensorflow/tensorflow/python/framework/*.proto
, third_party/tensorflow/tensorflow/python/keras/protobuf/*.proto
@ -57,11 +65,13 @@ extra-source-files: third_party/tensorflow/tensorflow/compiler/jit/*.proto
, third_party/tensorflow/tensorflow/python/tpu/*.proto
, third_party/tensorflow/tensorflow/python/training/*.proto
, third_party/tensorflow/tensorflow/python/util/protobuf/*.proto
, third_party/tensorflow/tensorflow/security/fuzzing/*.proto
, third_party/tensorflow/tensorflow/stream_executor/*.proto
, third_party/tensorflow/tensorflow/security/fuzzing/cc/*.proto
, third_party/tensorflow/tensorflow/tools/api/lib/*.proto
, third_party/tensorflow/tensorflow/tools/proto_text/*.proto
, third_party/tensorflow/tensorflow/tsl/distributed_runtime/coordination/*.proto
, third_party/tensorflow/tensorflow/tsl/distributed_runtime/rpc/*.proto
, third_party/tensorflow/tensorflow/tsl/profiler/protobuf/*.proto
, third_party/tensorflow/tensorflow/tsl/protobuf/*.proto
library
exposed-modules: Proto.Tensorflow.Compiler.Jit.XlaActivity
@ -70,6 +80,10 @@ library
, Proto.Tensorflow.Compiler.Jit.XlaCompilationCache_Fields
, Proto.Tensorflow.Compiler.Mlir.Lite.Quantization.QuantizationInfo
, Proto.Tensorflow.Compiler.Mlir.Lite.Quantization.QuantizationInfo_Fields
, Proto.Tensorflow.Compiler.Mlir.Quantization.Stablehlo.QuantizationOptions
, Proto.Tensorflow.Compiler.Mlir.Quantization.Stablehlo.QuantizationOptions_Fields
, Proto.Tensorflow.Compiler.Mlir.Quantization.Tensorflow.ExportedModel
, Proto.Tensorflow.Compiler.Mlir.Quantization.Tensorflow.ExportedModel_Fields
, Proto.Tensorflow.Compiler.Mlir.Quantization.Tensorflow.QuantizationOptions
, Proto.Tensorflow.Compiler.Mlir.Quantization.Tensorflow.QuantizationOptions_Fields
, Proto.Tensorflow.Compiler.Mlir.Tfrt.Analysis.Analysis
@ -82,10 +96,16 @@ library
, Proto.Tensorflow.Compiler.Tf2xla.HostComputeMetadata_Fields
, Proto.Tensorflow.Compiler.Tf2xla.Kernels.Callback
, Proto.Tensorflow.Compiler.Tf2xla.Kernels.Callback_Fields
, Proto.Tensorflow.Compiler.Tf2xla.Support
, Proto.Tensorflow.Compiler.Tf2xla.Support_Fields
, Proto.Tensorflow.Compiler.Tf2xla.Tf2xla
, Proto.Tensorflow.Compiler.Tf2xla.Tf2xla_Fields
, Proto.Tensorflow.Compiler.Xla.AutotuneResults
, Proto.Tensorflow.Compiler.Xla.AutotuneResults_Fields
, Proto.Tensorflow.Compiler.Xla.Mlir.Tools.MlirReplay.Public.CompilerTrace
, Proto.Tensorflow.Compiler.Xla.Mlir.Tools.MlirReplay.Public.CompilerTrace_Fields
, Proto.Tensorflow.Compiler.Xla.Mlir.Tools.MlirReplay.Public.ExecutionTrace
, Proto.Tensorflow.Compiler.Xla.Mlir.Tools.MlirReplay.Public.ExecutionTrace_Fields
, Proto.Tensorflow.Compiler.Xla.Pjrt.CompileOptions
, Proto.Tensorflow.Compiler.Xla.Pjrt.CompileOptions_Fields
, Proto.Tensorflow.Compiler.Xla.Pjrt.Distributed.Protocol
, Proto.Tensorflow.Compiler.Xla.Pjrt.Distributed.Protocol_Fields
, Proto.Tensorflow.Compiler.Xla.Python.TpuDriver.TpuDriver
@ -94,8 +114,18 @@ library
, Proto.Tensorflow.Compiler.Xla.Python.TpuDriver.TpuService_Fields
, Proto.Tensorflow.Compiler.Xla.Rpc.XlaService
, Proto.Tensorflow.Compiler.Xla.Rpc.XlaService_Fields
, Proto.Tensorflow.Compiler.Xla.Runtime.Runner.Runner
, Proto.Tensorflow.Compiler.Xla.Runtime.Runner.Runner_Fields
, Proto.Tensorflow.Compiler.Xla.Service.Cpu.BackendConfig
, Proto.Tensorflow.Compiler.Xla.Service.Cpu.BackendConfig_Fields
, Proto.Tensorflow.Compiler.Xla.Service.Cpu.Executable
, Proto.Tensorflow.Compiler.Xla.Service.Cpu.Executable_Fields
, Proto.Tensorflow.Compiler.Xla.Service.Cpu.XlaFramework
, Proto.Tensorflow.Compiler.Xla.Service.Cpu.XlaFramework_Fields
, Proto.Tensorflow.Compiler.Xla.Service.Gpu.BackendConfigs
, Proto.Tensorflow.Compiler.Xla.Service.Gpu.BackendConfigs_Fields
, Proto.Tensorflow.Compiler.Xla.Service.Gpu.Executable
, Proto.Tensorflow.Compiler.Xla.Service.Gpu.Executable_Fields
, Proto.Tensorflow.Compiler.Xla.Service.Gpu.GpuAutotuning
, Proto.Tensorflow.Compiler.Xla.Service.Gpu.GpuAutotuning_Fields
, Proto.Tensorflow.Compiler.Xla.Service.Hlo
@ -104,8 +134,14 @@ library
, Proto.Tensorflow.Compiler.Xla.Service.HloExecutionProfileData_Fields
, Proto.Tensorflow.Compiler.Xla.Service.HloProfilePrinterData
, Proto.Tensorflow.Compiler.Xla.Service.HloProfilePrinterData_Fields
, Proto.Tensorflow.Compiler.Xla.Service.Metrics
, Proto.Tensorflow.Compiler.Xla.Service.Metrics_Fields
, Proto.Tensorflow.Compiler.Xla.Service.TestCompilationEnvironment
, Proto.Tensorflow.Compiler.Xla.Service.TestCompilationEnvironment_Fields
, Proto.Tensorflow.Compiler.Xla.StreamExecutor.DeviceDescription
, Proto.Tensorflow.Compiler.Xla.StreamExecutor.DeviceDescription_Fields
, Proto.Tensorflow.Compiler.Xla.StreamExecutor.Dnn
, Proto.Tensorflow.Compiler.Xla.StreamExecutor.Dnn_Fields
, Proto.Tensorflow.Compiler.Xla.Tools.RunHloModule
, Proto.Tensorflow.Compiler.Xla.Tools.RunHloModule_Fields
, Proto.Tensorflow.Compiler.Xla.Xla
@ -114,8 +150,6 @@ library
, Proto.Tensorflow.Compiler.Xla.XlaData_Fields
, Proto.Tensorflow.Compiler.Xrt.Xrt
, Proto.Tensorflow.Compiler.Xrt.Xrt_Fields
, Proto.Tensorflow.Core.Data.Dataset
, Proto.Tensorflow.Core.Data.Dataset_Fields
, Proto.Tensorflow.Core.Data.Service.Common
, Proto.Tensorflow.Core.Data.Service.Common_Fields
, Proto.Tensorflow.Core.Data.Service.Dispatcher
@ -144,6 +178,8 @@ library
, Proto.Tensorflow.Core.Framework.AttrValue_Fields
, Proto.Tensorflow.Core.Framework.CostGraph
, Proto.Tensorflow.Core.Framework.CostGraph_Fields
, Proto.Tensorflow.Core.Framework.Dataset
, Proto.Tensorflow.Core.Framework.Dataset_Fields
, Proto.Tensorflow.Core.Framework.DatasetMetadata
, Proto.Tensorflow.Core.Framework.DatasetMetadata_Fields
, Proto.Tensorflow.Core.Framework.DatasetOptions
@ -168,6 +204,8 @@ library
, Proto.Tensorflow.Core.Framework.NodeDef_Fields
, Proto.Tensorflow.Core.Framework.OpDef
, Proto.Tensorflow.Core.Framework.OpDef_Fields
, Proto.Tensorflow.Core.Framework.OptimizedFunctionGraph
, Proto.Tensorflow.Core.Framework.OptimizedFunctionGraph_Fields
, Proto.Tensorflow.Core.Framework.ReaderBase
, Proto.Tensorflow.Core.Framework.ReaderBase_Fields
, Proto.Tensorflow.Core.Framework.ResourceHandle
@ -190,6 +228,8 @@ library
, Proto.Tensorflow.Core.Framework.Variable_Fields
, Proto.Tensorflow.Core.Framework.Versions
, Proto.Tensorflow.Core.Framework.Versions_Fields
, Proto.Tensorflow.Core.Function.Polymorphism.FunctionType
, Proto.Tensorflow.Core.Function.Polymorphism.FunctionType_Fields
, Proto.Tensorflow.Core.Function.TraceType.DefaultTypes
, Proto.Tensorflow.Core.Function.TraceType.DefaultTypes_Fields
, Proto.Tensorflow.Core.Function.TraceType.Serialization
@ -246,6 +286,8 @@ library
, Proto.Tensorflow.Core.Profiler.Protobuf.TfStats_Fields
, Proto.Tensorflow.Core.Profiler.Protobuf.Tfstreamz
, Proto.Tensorflow.Core.Profiler.Protobuf.Tfstreamz_Fields
, Proto.Tensorflow.Core.Profiler.Protobuf.Topology
, Proto.Tensorflow.Core.Profiler.Protobuf.Topology_Fields
, Proto.Tensorflow.Core.Profiler.Protobuf.TraceEvents
, Proto.Tensorflow.Core.Profiler.Protobuf.TraceEvents_Fields
, Proto.Tensorflow.Core.Profiler.Protobuf.Xplane
@ -270,10 +312,6 @@ library
, Proto.Tensorflow.Core.Protobuf.ControlFlow_Fields
, Proto.Tensorflow.Core.Protobuf.ConvAutotuning
, Proto.Tensorflow.Core.Protobuf.ConvAutotuning_Fields
, Proto.Tensorflow.Core.Protobuf.CoordinationConfig
, Proto.Tensorflow.Core.Protobuf.CoordinationConfig_Fields
, Proto.Tensorflow.Core.Protobuf.CoordinationService
, Proto.Tensorflow.Core.Protobuf.CoordinationService_Fields
, Proto.Tensorflow.Core.Protobuf.CorePlatformPayloads
, Proto.Tensorflow.Core.Protobuf.CorePlatformPayloads_Fields
, Proto.Tensorflow.Core.Protobuf.CriticalSection
@ -288,8 +326,6 @@ library
, Proto.Tensorflow.Core.Protobuf.DeviceFilters_Fields
, Proto.Tensorflow.Core.Protobuf.DeviceProperties
, Proto.Tensorflow.Core.Protobuf.DeviceProperties_Fields
, Proto.Tensorflow.Core.Protobuf.DistributedRuntimePayloads
, Proto.Tensorflow.Core.Protobuf.DistributedRuntimePayloads_Fields
, Proto.Tensorflow.Core.Protobuf.EagerService
, Proto.Tensorflow.Core.Protobuf.EagerService_Fields
, Proto.Tensorflow.Core.Protobuf.ErrorCodes
@ -314,6 +350,8 @@ library
, Proto.Tensorflow.Core.Protobuf.ReplayLog_Fields
, Proto.Tensorflow.Core.Protobuf.RewriterConfig
, Proto.Tensorflow.Core.Protobuf.RewriterConfig_Fields
, Proto.Tensorflow.Core.Protobuf.RpcOptions
, Proto.Tensorflow.Core.Protobuf.RpcOptions_Fields
, Proto.Tensorflow.Core.Protobuf.SavedModel
, Proto.Tensorflow.Core.Protobuf.SavedModel_Fields
, Proto.Tensorflow.Core.Protobuf.SavedObjectGraph
@ -324,8 +362,6 @@ library
, Proto.Tensorflow.Core.Protobuf.ServiceConfig_Fields
, Proto.Tensorflow.Core.Protobuf.Snapshot
, Proto.Tensorflow.Core.Protobuf.Snapshot_Fields
, Proto.Tensorflow.Core.Protobuf.Status
, Proto.Tensorflow.Core.Protobuf.Status_Fields
, Proto.Tensorflow.Core.Protobuf.Struct
, Proto.Tensorflow.Core.Protobuf.Struct_Fields
, Proto.Tensorflow.Core.Protobuf.TensorBundle
@ -374,6 +410,8 @@ library
, Proto.Tensorflow.Core.Util.ExampleProtoFastParsingTest_Fields
, Proto.Tensorflow.Core.Util.MemmappedFileSystem
, Proto.Tensorflow.Core.Util.MemmappedFileSystem_Fields
, Proto.Tensorflow.Core.Util.Quantization.UniformQuantOpsAttr
, Proto.Tensorflow.Core.Util.Quantization.UniformQuantOpsAttr_Fields
, Proto.Tensorflow.Core.Util.SavedTensorSlice
, Proto.Tensorflow.Core.Util.SavedTensorSlice_Fields
, Proto.Tensorflow.Core.Util.TestLog
@ -394,6 +432,8 @@ library
, Proto.Tensorflow.Lite.Toco.TocoFlags_Fields
, Proto.Tensorflow.Lite.Toco.Types
, Proto.Tensorflow.Lite.Toco.Types_Fields
, Proto.Tensorflow.Lite.Tools.Delegates.Compatibility.Protos.CompatibilityResult
, Proto.Tensorflow.Lite.Tools.Delegates.Compatibility.Protos.CompatibilityResult_Fields
, Proto.Tensorflow.Lite.Tools.Evaluation.Proto.EvaluationConfig
, Proto.Tensorflow.Lite.Tools.Evaluation.Proto.EvaluationConfig_Fields
, Proto.Tensorflow.Lite.Tools.Evaluation.Proto.EvaluationStages
@ -402,6 +442,10 @@ library
, Proto.Tensorflow.Lite.Tools.Evaluation.Proto.PreprocessingSteps_Fields
, Proto.Tensorflow.Python.Framework.CppShapeInference
, Proto.Tensorflow.Python.Framework.CppShapeInference_Fields
, Proto.Tensorflow.Python.Framework.KytheMetadata
, Proto.Tensorflow.Python.Framework.KytheMetadata_Fields
, Proto.Tensorflow.Python.Framework.OpRegOffset
, Proto.Tensorflow.Python.Framework.OpRegOffset_Fields
, Proto.Tensorflow.Python.Keras.Protobuf.ProjectorConfig
, Proto.Tensorflow.Python.Keras.Protobuf.ProjectorConfig_Fields
, Proto.Tensorflow.Python.Keras.Protobuf.SavedMetadata
@ -416,20 +460,60 @@ library
, Proto.Tensorflow.Python.Training.CheckpointState_Fields
, Proto.Tensorflow.Python.Util.Protobuf.CompareTest
, Proto.Tensorflow.Python.Util.Protobuf.CompareTest_Fields
, Proto.Tensorflow.Security.Fuzzing.CheckpointReaderFuzzInput
, Proto.Tensorflow.Security.Fuzzing.CheckpointReaderFuzzInput_Fields
, Proto.Tensorflow.StreamExecutor.Dnn
, Proto.Tensorflow.StreamExecutor.Dnn_Fields
, Proto.Tensorflow.Security.Fuzzing.Cc.CheckpointReaderFuzzInput
, Proto.Tensorflow.Security.Fuzzing.Cc.CheckpointReaderFuzzInput_Fields
, Proto.Tensorflow.Tools.Api.Lib.ApiObjects
, Proto.Tensorflow.Tools.Api.Lib.ApiObjects_Fields
, Proto.Tensorflow.Tools.ProtoText.Test
, Proto.Tensorflow.Tools.ProtoText.Test_Fields
, Proto.Tensorflow.Tsl.DistributedRuntime.Coordination.TestDevice
, Proto.Tensorflow.Tsl.DistributedRuntime.Coordination.TestDevice_Fields
, Proto.Tensorflow.Tsl.DistributedRuntime.Rpc.TestRequest
, Proto.Tensorflow.Tsl.DistributedRuntime.Rpc.TestRequest_Fields
, Proto.Tensorflow.Tsl.Profiler.Protobuf.Profile
, Proto.Tensorflow.Tsl.Profiler.Protobuf.Profile_Fields
, Proto.Tensorflow.Tsl.Profiler.Protobuf.ProfilerAnalysis
, Proto.Tensorflow.Tsl.Profiler.Protobuf.ProfilerAnalysis_Fields
, Proto.Tensorflow.Tsl.Profiler.Protobuf.ProfilerOptions
, Proto.Tensorflow.Tsl.Profiler.Protobuf.ProfilerOptions_Fields
, Proto.Tensorflow.Tsl.Profiler.Protobuf.ProfilerService
, Proto.Tensorflow.Tsl.Profiler.Protobuf.ProfilerService_Fields
, Proto.Tensorflow.Tsl.Profiler.Protobuf.ProfilerServiceMonitorResult
, Proto.Tensorflow.Tsl.Profiler.Protobuf.ProfilerServiceMonitorResult_Fields
, Proto.Tensorflow.Tsl.Profiler.Protobuf.TraceEvents
, Proto.Tensorflow.Tsl.Profiler.Protobuf.TraceEvents_Fields
, Proto.Tensorflow.Tsl.Profiler.Protobuf.Xplane
, Proto.Tensorflow.Tsl.Profiler.Protobuf.Xplane_Fields
, Proto.Tensorflow.Tsl.Protobuf.Autotuning
, Proto.Tensorflow.Tsl.Protobuf.Autotuning_Fields
, Proto.Tensorflow.Tsl.Protobuf.BfcMemoryMap
, Proto.Tensorflow.Tsl.Protobuf.BfcMemoryMap_Fields
, Proto.Tensorflow.Tsl.Protobuf.CoordinationConfig
, Proto.Tensorflow.Tsl.Protobuf.CoordinationConfig_Fields
, Proto.Tensorflow.Tsl.Protobuf.CoordinationService
, Proto.Tensorflow.Tsl.Protobuf.CoordinationService_Fields
, Proto.Tensorflow.Tsl.Protobuf.DistributedRuntimePayloads
, Proto.Tensorflow.Tsl.Protobuf.DistributedRuntimePayloads_Fields
, Proto.Tensorflow.Tsl.Protobuf.Dnn
, Proto.Tensorflow.Tsl.Protobuf.Dnn_Fields
, Proto.Tensorflow.Tsl.Protobuf.ErrorCodes
, Proto.Tensorflow.Tsl.Protobuf.ErrorCodes_Fields
, Proto.Tensorflow.Tsl.Protobuf.Histogram
, Proto.Tensorflow.Tsl.Protobuf.Histogram_Fields
, Proto.Tensorflow.Tsl.Protobuf.RpcOptions
, Proto.Tensorflow.Tsl.Protobuf.RpcOptions_Fields
, Proto.Tensorflow.Tsl.Protobuf.TestLog
, Proto.Tensorflow.Tsl.Protobuf.TestLog_Fields
autogen-modules: Proto.Tensorflow.Compiler.Jit.XlaActivity
, Proto.Tensorflow.Compiler.Jit.XlaActivity_Fields
, Proto.Tensorflow.Compiler.Jit.XlaCompilationCache
, Proto.Tensorflow.Compiler.Jit.XlaCompilationCache_Fields
, Proto.Tensorflow.Compiler.Mlir.Lite.Quantization.QuantizationInfo
, Proto.Tensorflow.Compiler.Mlir.Lite.Quantization.QuantizationInfo_Fields
, Proto.Tensorflow.Compiler.Mlir.Quantization.Stablehlo.QuantizationOptions
, Proto.Tensorflow.Compiler.Mlir.Quantization.Stablehlo.QuantizationOptions_Fields
, Proto.Tensorflow.Compiler.Mlir.Quantization.Tensorflow.ExportedModel
, Proto.Tensorflow.Compiler.Mlir.Quantization.Tensorflow.ExportedModel_Fields
, Proto.Tensorflow.Compiler.Mlir.Quantization.Tensorflow.QuantizationOptions
, Proto.Tensorflow.Compiler.Mlir.Quantization.Tensorflow.QuantizationOptions_Fields
, Proto.Tensorflow.Compiler.Mlir.Tfrt.Analysis.Analysis
@ -442,10 +526,16 @@ library
, Proto.Tensorflow.Compiler.Tf2xla.HostComputeMetadata_Fields
, Proto.Tensorflow.Compiler.Tf2xla.Kernels.Callback
, Proto.Tensorflow.Compiler.Tf2xla.Kernels.Callback_Fields
, Proto.Tensorflow.Compiler.Tf2xla.Support
, Proto.Tensorflow.Compiler.Tf2xla.Support_Fields
, Proto.Tensorflow.Compiler.Tf2xla.Tf2xla
, Proto.Tensorflow.Compiler.Tf2xla.Tf2xla_Fields
, Proto.Tensorflow.Compiler.Xla.AutotuneResults
, Proto.Tensorflow.Compiler.Xla.AutotuneResults_Fields
, Proto.Tensorflow.Compiler.Xla.Mlir.Tools.MlirReplay.Public.CompilerTrace
, Proto.Tensorflow.Compiler.Xla.Mlir.Tools.MlirReplay.Public.CompilerTrace_Fields
, Proto.Tensorflow.Compiler.Xla.Mlir.Tools.MlirReplay.Public.ExecutionTrace
, Proto.Tensorflow.Compiler.Xla.Mlir.Tools.MlirReplay.Public.ExecutionTrace_Fields
, Proto.Tensorflow.Compiler.Xla.Pjrt.CompileOptions
, Proto.Tensorflow.Compiler.Xla.Pjrt.CompileOptions_Fields
, Proto.Tensorflow.Compiler.Xla.Pjrt.Distributed.Protocol
, Proto.Tensorflow.Compiler.Xla.Pjrt.Distributed.Protocol_Fields
, Proto.Tensorflow.Compiler.Xla.Python.TpuDriver.TpuDriver
@ -454,8 +544,18 @@ library
, Proto.Tensorflow.Compiler.Xla.Python.TpuDriver.TpuService_Fields
, Proto.Tensorflow.Compiler.Xla.Rpc.XlaService
, Proto.Tensorflow.Compiler.Xla.Rpc.XlaService_Fields
, Proto.Tensorflow.Compiler.Xla.Runtime.Runner.Runner
, Proto.Tensorflow.Compiler.Xla.Runtime.Runner.Runner_Fields
, Proto.Tensorflow.Compiler.Xla.Service.Cpu.BackendConfig
, Proto.Tensorflow.Compiler.Xla.Service.Cpu.BackendConfig_Fields
, Proto.Tensorflow.Compiler.Xla.Service.Cpu.Executable
, Proto.Tensorflow.Compiler.Xla.Service.Cpu.Executable_Fields
, Proto.Tensorflow.Compiler.Xla.Service.Cpu.XlaFramework
, Proto.Tensorflow.Compiler.Xla.Service.Cpu.XlaFramework_Fields
, Proto.Tensorflow.Compiler.Xla.Service.Gpu.BackendConfigs
, Proto.Tensorflow.Compiler.Xla.Service.Gpu.BackendConfigs_Fields
, Proto.Tensorflow.Compiler.Xla.Service.Gpu.Executable
, Proto.Tensorflow.Compiler.Xla.Service.Gpu.Executable_Fields
, Proto.Tensorflow.Compiler.Xla.Service.Gpu.GpuAutotuning
, Proto.Tensorflow.Compiler.Xla.Service.Gpu.GpuAutotuning_Fields
, Proto.Tensorflow.Compiler.Xla.Service.Hlo
@ -464,8 +564,14 @@ library
, Proto.Tensorflow.Compiler.Xla.Service.HloExecutionProfileData_Fields
, Proto.Tensorflow.Compiler.Xla.Service.HloProfilePrinterData
, Proto.Tensorflow.Compiler.Xla.Service.HloProfilePrinterData_Fields
, Proto.Tensorflow.Compiler.Xla.Service.Metrics
, Proto.Tensorflow.Compiler.Xla.Service.Metrics_Fields
, Proto.Tensorflow.Compiler.Xla.Service.TestCompilationEnvironment
, Proto.Tensorflow.Compiler.Xla.Service.TestCompilationEnvironment_Fields
, Proto.Tensorflow.Compiler.Xla.StreamExecutor.DeviceDescription
, Proto.Tensorflow.Compiler.Xla.StreamExecutor.DeviceDescription_Fields
, Proto.Tensorflow.Compiler.Xla.StreamExecutor.Dnn
, Proto.Tensorflow.Compiler.Xla.StreamExecutor.Dnn_Fields
, Proto.Tensorflow.Compiler.Xla.Tools.RunHloModule
, Proto.Tensorflow.Compiler.Xla.Tools.RunHloModule_Fields
, Proto.Tensorflow.Compiler.Xla.Xla
@ -474,8 +580,6 @@ library
, Proto.Tensorflow.Compiler.Xla.XlaData_Fields
, Proto.Tensorflow.Compiler.Xrt.Xrt
, Proto.Tensorflow.Compiler.Xrt.Xrt_Fields
, Proto.Tensorflow.Core.Data.Dataset
, Proto.Tensorflow.Core.Data.Dataset_Fields
, Proto.Tensorflow.Core.Data.Service.Common
, Proto.Tensorflow.Core.Data.Service.Common_Fields
, Proto.Tensorflow.Core.Data.Service.Dispatcher
@ -504,6 +608,8 @@ library
, Proto.Tensorflow.Core.Framework.AttrValue_Fields
, Proto.Tensorflow.Core.Framework.CostGraph
, Proto.Tensorflow.Core.Framework.CostGraph_Fields
, Proto.Tensorflow.Core.Framework.Dataset
, Proto.Tensorflow.Core.Framework.Dataset_Fields
, Proto.Tensorflow.Core.Framework.DatasetMetadata
, Proto.Tensorflow.Core.Framework.DatasetMetadata_Fields
, Proto.Tensorflow.Core.Framework.DatasetOptions
@ -528,6 +634,8 @@ library
, Proto.Tensorflow.Core.Framework.NodeDef_Fields
, Proto.Tensorflow.Core.Framework.OpDef
, Proto.Tensorflow.Core.Framework.OpDef_Fields
, Proto.Tensorflow.Core.Framework.OptimizedFunctionGraph
, Proto.Tensorflow.Core.Framework.OptimizedFunctionGraph_Fields
, Proto.Tensorflow.Core.Framework.ReaderBase
, Proto.Tensorflow.Core.Framework.ReaderBase_Fields
, Proto.Tensorflow.Core.Framework.ResourceHandle
@ -550,6 +658,8 @@ library
, Proto.Tensorflow.Core.Framework.Variable_Fields
, Proto.Tensorflow.Core.Framework.Versions
, Proto.Tensorflow.Core.Framework.Versions_Fields
, Proto.Tensorflow.Core.Function.Polymorphism.FunctionType
, Proto.Tensorflow.Core.Function.Polymorphism.FunctionType_Fields
, Proto.Tensorflow.Core.Function.TraceType.DefaultTypes
, Proto.Tensorflow.Core.Function.TraceType.DefaultTypes_Fields
, Proto.Tensorflow.Core.Function.TraceType.Serialization
@ -606,6 +716,8 @@ library
, Proto.Tensorflow.Core.Profiler.Protobuf.TfStats_Fields
, Proto.Tensorflow.Core.Profiler.Protobuf.Tfstreamz
, Proto.Tensorflow.Core.Profiler.Protobuf.Tfstreamz_Fields
, Proto.Tensorflow.Core.Profiler.Protobuf.Topology
, Proto.Tensorflow.Core.Profiler.Protobuf.Topology_Fields
, Proto.Tensorflow.Core.Profiler.Protobuf.TraceEvents
, Proto.Tensorflow.Core.Profiler.Protobuf.TraceEvents_Fields
, Proto.Tensorflow.Core.Profiler.Protobuf.Xplane
@ -630,10 +742,6 @@ library
, Proto.Tensorflow.Core.Protobuf.ControlFlow_Fields
, Proto.Tensorflow.Core.Protobuf.ConvAutotuning
, Proto.Tensorflow.Core.Protobuf.ConvAutotuning_Fields
, Proto.Tensorflow.Core.Protobuf.CoordinationConfig
, Proto.Tensorflow.Core.Protobuf.CoordinationConfig_Fields
, Proto.Tensorflow.Core.Protobuf.CoordinationService
, Proto.Tensorflow.Core.Protobuf.CoordinationService_Fields
, Proto.Tensorflow.Core.Protobuf.CorePlatformPayloads
, Proto.Tensorflow.Core.Protobuf.CorePlatformPayloads_Fields
, Proto.Tensorflow.Core.Protobuf.CriticalSection
@ -648,8 +756,6 @@ library
, Proto.Tensorflow.Core.Protobuf.DeviceFilters_Fields
, Proto.Tensorflow.Core.Protobuf.DeviceProperties
, Proto.Tensorflow.Core.Protobuf.DeviceProperties_Fields
, Proto.Tensorflow.Core.Protobuf.DistributedRuntimePayloads
, Proto.Tensorflow.Core.Protobuf.DistributedRuntimePayloads_Fields
, Proto.Tensorflow.Core.Protobuf.EagerService
, Proto.Tensorflow.Core.Protobuf.EagerService_Fields
, Proto.Tensorflow.Core.Protobuf.ErrorCodes
@ -674,6 +780,8 @@ library
, Proto.Tensorflow.Core.Protobuf.ReplayLog_Fields
, Proto.Tensorflow.Core.Protobuf.RewriterConfig
, Proto.Tensorflow.Core.Protobuf.RewriterConfig_Fields
, Proto.Tensorflow.Core.Protobuf.RpcOptions
, Proto.Tensorflow.Core.Protobuf.RpcOptions_Fields
, Proto.Tensorflow.Core.Protobuf.SavedModel
, Proto.Tensorflow.Core.Protobuf.SavedModel_Fields
, Proto.Tensorflow.Core.Protobuf.SavedObjectGraph
@ -684,8 +792,6 @@ library
, Proto.Tensorflow.Core.Protobuf.ServiceConfig_Fields
, Proto.Tensorflow.Core.Protobuf.Snapshot
, Proto.Tensorflow.Core.Protobuf.Snapshot_Fields
, Proto.Tensorflow.Core.Protobuf.Status
, Proto.Tensorflow.Core.Protobuf.Status_Fields
, Proto.Tensorflow.Core.Protobuf.Struct
, Proto.Tensorflow.Core.Protobuf.Struct_Fields
, Proto.Tensorflow.Core.Protobuf.TensorBundle
@ -734,6 +840,8 @@ library
, Proto.Tensorflow.Core.Util.ExampleProtoFastParsingTest_Fields
, Proto.Tensorflow.Core.Util.MemmappedFileSystem
, Proto.Tensorflow.Core.Util.MemmappedFileSystem_Fields
, Proto.Tensorflow.Core.Util.Quantization.UniformQuantOpsAttr
, Proto.Tensorflow.Core.Util.Quantization.UniformQuantOpsAttr_Fields
, Proto.Tensorflow.Core.Util.SavedTensorSlice
, Proto.Tensorflow.Core.Util.SavedTensorSlice_Fields
, Proto.Tensorflow.Core.Util.TestLog
@ -754,6 +862,8 @@ library
, Proto.Tensorflow.Lite.Toco.TocoFlags_Fields
, Proto.Tensorflow.Lite.Toco.Types
, Proto.Tensorflow.Lite.Toco.Types_Fields
, Proto.Tensorflow.Lite.Tools.Delegates.Compatibility.Protos.CompatibilityResult
, Proto.Tensorflow.Lite.Tools.Delegates.Compatibility.Protos.CompatibilityResult_Fields
, Proto.Tensorflow.Lite.Tools.Evaluation.Proto.EvaluationConfig
, Proto.Tensorflow.Lite.Tools.Evaluation.Proto.EvaluationConfig_Fields
, Proto.Tensorflow.Lite.Tools.Evaluation.Proto.EvaluationStages
@ -762,6 +872,10 @@ library
, Proto.Tensorflow.Lite.Tools.Evaluation.Proto.PreprocessingSteps_Fields
, Proto.Tensorflow.Python.Framework.CppShapeInference
, Proto.Tensorflow.Python.Framework.CppShapeInference_Fields
, Proto.Tensorflow.Python.Framework.KytheMetadata
, Proto.Tensorflow.Python.Framework.KytheMetadata_Fields
, Proto.Tensorflow.Python.Framework.OpRegOffset
, Proto.Tensorflow.Python.Framework.OpRegOffset_Fields
, Proto.Tensorflow.Python.Keras.Protobuf.ProjectorConfig
, Proto.Tensorflow.Python.Keras.Protobuf.ProjectorConfig_Fields
, Proto.Tensorflow.Python.Keras.Protobuf.SavedMetadata
@ -776,14 +890,50 @@ library
, Proto.Tensorflow.Python.Training.CheckpointState_Fields
, Proto.Tensorflow.Python.Util.Protobuf.CompareTest
, Proto.Tensorflow.Python.Util.Protobuf.CompareTest_Fields
, Proto.Tensorflow.Security.Fuzzing.CheckpointReaderFuzzInput
, Proto.Tensorflow.Security.Fuzzing.CheckpointReaderFuzzInput_Fields
, Proto.Tensorflow.StreamExecutor.Dnn
, Proto.Tensorflow.StreamExecutor.Dnn_Fields
, Proto.Tensorflow.Security.Fuzzing.Cc.CheckpointReaderFuzzInput
, Proto.Tensorflow.Security.Fuzzing.Cc.CheckpointReaderFuzzInput_Fields
, Proto.Tensorflow.Tools.Api.Lib.ApiObjects
, Proto.Tensorflow.Tools.Api.Lib.ApiObjects_Fields
, Proto.Tensorflow.Tools.ProtoText.Test
, Proto.Tensorflow.Tools.ProtoText.Test_Fields
, Proto.Tensorflow.Tsl.DistributedRuntime.Coordination.TestDevice
, Proto.Tensorflow.Tsl.DistributedRuntime.Coordination.TestDevice_Fields
, Proto.Tensorflow.Tsl.DistributedRuntime.Rpc.TestRequest
, Proto.Tensorflow.Tsl.DistributedRuntime.Rpc.TestRequest_Fields
, Proto.Tensorflow.Tsl.Profiler.Protobuf.Profile
, Proto.Tensorflow.Tsl.Profiler.Protobuf.Profile_Fields
, Proto.Tensorflow.Tsl.Profiler.Protobuf.ProfilerAnalysis
, Proto.Tensorflow.Tsl.Profiler.Protobuf.ProfilerAnalysis_Fields
, Proto.Tensorflow.Tsl.Profiler.Protobuf.ProfilerOptions
, Proto.Tensorflow.Tsl.Profiler.Protobuf.ProfilerOptions_Fields
, Proto.Tensorflow.Tsl.Profiler.Protobuf.ProfilerService
, Proto.Tensorflow.Tsl.Profiler.Protobuf.ProfilerService_Fields
, Proto.Tensorflow.Tsl.Profiler.Protobuf.ProfilerServiceMonitorResult
, Proto.Tensorflow.Tsl.Profiler.Protobuf.ProfilerServiceMonitorResult_Fields
, Proto.Tensorflow.Tsl.Profiler.Protobuf.TraceEvents
, Proto.Tensorflow.Tsl.Profiler.Protobuf.TraceEvents_Fields
, Proto.Tensorflow.Tsl.Profiler.Protobuf.Xplane
, Proto.Tensorflow.Tsl.Profiler.Protobuf.Xplane_Fields
, Proto.Tensorflow.Tsl.Protobuf.Autotuning
, Proto.Tensorflow.Tsl.Protobuf.Autotuning_Fields
, Proto.Tensorflow.Tsl.Protobuf.BfcMemoryMap
, Proto.Tensorflow.Tsl.Protobuf.BfcMemoryMap_Fields
, Proto.Tensorflow.Tsl.Protobuf.CoordinationConfig
, Proto.Tensorflow.Tsl.Protobuf.CoordinationConfig_Fields
, Proto.Tensorflow.Tsl.Protobuf.CoordinationService
, Proto.Tensorflow.Tsl.Protobuf.CoordinationService_Fields
, Proto.Tensorflow.Tsl.Protobuf.DistributedRuntimePayloads
, Proto.Tensorflow.Tsl.Protobuf.DistributedRuntimePayloads_Fields
, Proto.Tensorflow.Tsl.Protobuf.Dnn
, Proto.Tensorflow.Tsl.Protobuf.Dnn_Fields
, Proto.Tensorflow.Tsl.Protobuf.ErrorCodes
, Proto.Tensorflow.Tsl.Protobuf.ErrorCodes_Fields
, Proto.Tensorflow.Tsl.Protobuf.Histogram
, Proto.Tensorflow.Tsl.Protobuf.Histogram_Fields
, Proto.Tensorflow.Tsl.Protobuf.RpcOptions
, Proto.Tensorflow.Tsl.Protobuf.RpcOptions_Fields
, Proto.Tensorflow.Tsl.Protobuf.TestLog
, Proto.Tensorflow.Tsl.Protobuf.TestLog_Fields
build-depends: proto-lens == 0.7.*
, proto-lens-runtime == 0.7.*
, proto-lens-protobuf-types == 0.7.*

View file

@ -261,7 +261,7 @@ resolveOperation graph name = do
where
exception =
let msg = "Operation not found in graph: " <> (T.pack $ show name)
in TensorFlowException Raw.TF_INVALID_ARGUMENT msg
in TensorFlowException Raw.TSL_INVALID_ARGUMENT msg
-- Internal.
@ -325,7 +325,7 @@ checkStatus fn =
bracket Raw.newStatus Raw.deleteStatus $ \status -> do
result <- fn status
code <- Raw.getCode status
when (code /= Raw.TF_OK) $ do
when (code /= Raw.TSL_OK) $ do
msg <- T.decodeUtf8With T.lenientDecode <$>
(Raw.message status >>= B.packCString)
throwM $ TensorFlowException code msg
@ -372,4 +372,4 @@ getAllOpList =
when (p == nullPtr) (throwM exception)
return p
exception = TensorFlowException
Raw.TF_UNKNOWN "GetAllOpList failure, check logs"
Raw.TSL_UNKNOWN "GetAllOpList failure, check logs"

@ -1 +1 @@
Subproject commit fdfc646704c37bdf450525f6ced9d80df86e4993
Subproject commit 0db597d0d758aba578783b5bf46c889700a45085