Why Go
- Case Studies
  
  Common problems companies solve with Go
- Use Cases
  
  Stories about how and why companies use Go
- Security Policy
  
  How Go can help keep you secure by default
Learn
Docs
- Effective Go
  
  Tips for writing clear, performant, and idiomatic Go code
- Go User Manual
  
  A complete introduction to building software with Go
- Standard library
  
  Reference documentation for Go's standard library
- Release Notes
  
  Learn what's new in each Go release
Packages
Community
- Recorded Talks
  
  Videos from prior events
- Meetups
  
  Meet other local Go developers
- Conferences
  
  Learn and network with Go developers from around the world
- Go blog
  
  The Go project's official blog.
- Go project
  
  Get help and stay informed from Go
- Get connected

kernels

package

Go to main page

Versions in this module

v2

v2.14.1

Dec 8, 2023

v2.14.0

Dec 8, 2023

Changes in this version

type TPUHostTransferInfoProto

+ SerializedTransfers [][]byte

+ func (x *TPUHostTransferInfoProto) GetSerializedTransfers() [][]byte

v2.13.0

Dec 8, 2023

v2.12.0

Dec 8, 2023

v2.11.0

Dec 8, 2023

v2.10.1

Dec 8, 2023

v2.10.0

Dec 8, 2023

Changes in this version

+ const TpuCompilationCacheServiceExternal_GetTpuProgram_FullMethodName

+ var CompilationCacheFetchTarget_name = map[int32]string

+ var CompilationCacheFetchTarget_value = map[string]int32

+ var File_tensorflow_core_tpu_kernels_tpu_compilation_cache_common_proto protoreflect.FileDescriptor

+ var File_tensorflow_core_tpu_kernels_tpu_compilation_cache_proto protoreflect.FileDescriptor

+ var File_tensorflow_core_tpu_kernels_tpu_compile_proto protoreflect.FileDescriptor

+ var File_tensorflow_core_tpu_kernels_tpu_executable_info_proto protoreflect.FileDescriptor

+ var TPUHostTransferProto_TransferDirection_name = map[int32]string

+ var TPUHostTransferProto_TransferDirection_value = map[string]int32

+ var TpuCompilationCacheServiceExternal_ServiceDesc = grpc.ServiceDesc

+ func RegisterTpuCompilationCacheServiceExternalServer(s grpc.ServiceRegistrar, srv TpuCompilationCacheServiceExternalServer)

+ type CompilationCacheFetchTarget int32

+ const CompilationCacheFetchTarget_INVALID

+ const CompilationCacheFetchTarget_MAIN

+ const CompilationCacheFetchTarget_SHARDING

+ const CompilationCacheFetchTarget_UNSHARDING

+ func (CompilationCacheFetchTarget) Descriptor() protoreflect.EnumDescriptor

+ func (CompilationCacheFetchTarget) EnumDescriptor() ([]byte, []int)

+ func (CompilationCacheFetchTarget) Type() protoreflect.EnumType

+ func (x CompilationCacheFetchTarget) Enum() *CompilationCacheFetchTarget

+ func (x CompilationCacheFetchTarget) Number() protoreflect.EnumNumber

+ func (x CompilationCacheFetchTarget) String() string

+ type GetTpuProgramRequest struct

+ FetchTarget CompilationCacheFetchTarget

+ KeyOneof isGetTpuProgramRequest_KeyOneof

+ func (*GetTpuProgramRequest) Descriptor() ([]byte, []int)

+ func (*GetTpuProgramRequest) ProtoMessage()

+ func (m *GetTpuProgramRequest) GetKeyOneof() isGetTpuProgramRequest_KeyOneof

+ func (x *GetTpuProgramRequest) GetFetchTarget() CompilationCacheFetchTarget

+ func (x *GetTpuProgramRequest) GetKey() string

+ func (x *GetTpuProgramRequest) GetUidAndIndex() *TpuCompilationUidAndIndex

+ func (x *GetTpuProgramRequest) ProtoReflect() protoreflect.Message

+ func (x *GetTpuProgramRequest) Reset()

+ func (x *GetTpuProgramRequest) String() string

+ type GetTpuProgramRequest_Key struct

+ Key string

+ type GetTpuProgramRequest_UidAndIndex struct

+ UidAndIndex *TpuCompilationUidAndIndex

+ type GetTpuProgramResponseExternal struct

+ CompilerMetadata *GetTpuProgramResponseExternal_Blob

+ HostComputeMetadata *tf2xla.HostComputeMetadata

+ IsEmpty bool

+ MayModifyVariables bool

+ Proto *GetTpuProgramResponseExternal_Blob

+ func (*GetTpuProgramResponseExternal) Descriptor() ([]byte, []int)

+ func (*GetTpuProgramResponseExternal) ProtoMessage()

+ func (x *GetTpuProgramResponseExternal) GetCompilerMetadata() *GetTpuProgramResponseExternal_Blob

+ func (x *GetTpuProgramResponseExternal) GetHostComputeMetadata() *tf2xla.HostComputeMetadata

+ func (x *GetTpuProgramResponseExternal) GetIsEmpty() bool

+ func (x *GetTpuProgramResponseExternal) GetMayModifyVariables() bool

+ func (x *GetTpuProgramResponseExternal) GetProto() *GetTpuProgramResponseExternal_Blob

+ func (x *GetTpuProgramResponseExternal) ProtoReflect() protoreflect.Message

+ func (x *GetTpuProgramResponseExternal) Reset()

+ func (x *GetTpuProgramResponseExternal) String() string

+ type GetTpuProgramResponseExternal_Blob struct

+ Data []byte

+ func (*GetTpuProgramResponseExternal_Blob) Descriptor() ([]byte, []int)

+ func (*GetTpuProgramResponseExternal_Blob) ProtoMessage()

+ func (x *GetTpuProgramResponseExternal_Blob) GetData() []byte

+ func (x *GetTpuProgramResponseExternal_Blob) ProtoReflect() protoreflect.Message

+ func (x *GetTpuProgramResponseExternal_Blob) Reset()

+ func (x *GetTpuProgramResponseExternal_Blob) String() string

+ type TPUExecutableInfoProto struct

+ DeviceAssignment *data.DeviceAssignmentProto

+ DynamicOutputIndices []*TPUExecutableInfoProto_ShapeIndex

+ InputShapes []*data.ShapeProto

+ OutputShape *data.ShapeProto

+ OutputTensorShapes []*framework.TensorShapeProto

+ SessionModule *service.HloSnapshot

+ VariableIndices []*TPUExecutableInfoProto_UpdateIndexPair

+ func (*TPUExecutableInfoProto) Descriptor() ([]byte, []int)

+ func (*TPUExecutableInfoProto) ProtoMessage()

+ func (x *TPUExecutableInfoProto) GetDeviceAssignment() *data.DeviceAssignmentProto

+ func (x *TPUExecutableInfoProto) GetDynamicOutputIndices() []*TPUExecutableInfoProto_ShapeIndex

+ func (x *TPUExecutableInfoProto) GetInputShapes() []*data.ShapeProto

+ func (x *TPUExecutableInfoProto) GetOutputShape() *data.ShapeProto

+ func (x *TPUExecutableInfoProto) GetOutputTensorShapes() []*framework.TensorShapeProto

+ func (x *TPUExecutableInfoProto) GetSessionModule() *service.HloSnapshot

+ func (x *TPUExecutableInfoProto) GetVariableIndices() []*TPUExecutableInfoProto_UpdateIndexPair

+ func (x *TPUExecutableInfoProto) ProtoReflect() protoreflect.Message

+ func (x *TPUExecutableInfoProto) Reset()

+ func (x *TPUExecutableInfoProto) String() string

+ type TPUExecutableInfoProto_ShapeIndex struct

+ Index []int32

+ func (*TPUExecutableInfoProto_ShapeIndex) Descriptor() ([]byte, []int)

+ func (*TPUExecutableInfoProto_ShapeIndex) ProtoMessage()

+ func (x *TPUExecutableInfoProto_ShapeIndex) GetIndex() []int32

+ func (x *TPUExecutableInfoProto_ShapeIndex) ProtoReflect() protoreflect.Message

+ func (x *TPUExecutableInfoProto_ShapeIndex) Reset()

+ func (x *TPUExecutableInfoProto_ShapeIndex) String() string

+ type TPUExecutableInfoProto_UpdateIndexPair struct

+ Index int32

+ Updated bool

+ func (*TPUExecutableInfoProto_UpdateIndexPair) Descriptor() ([]byte, []int)

+ func (*TPUExecutableInfoProto_UpdateIndexPair) ProtoMessage()

+ func (x *TPUExecutableInfoProto_UpdateIndexPair) GetIndex() int32

+ func (x *TPUExecutableInfoProto_UpdateIndexPair) GetUpdated() bool

+ func (x *TPUExecutableInfoProto_UpdateIndexPair) ProtoReflect() protoreflect.Message

+ func (x *TPUExecutableInfoProto_UpdateIndexPair) Reset()

+ func (x *TPUExecutableInfoProto_UpdateIndexPair) String() string

+ type TPUHostTransferInfoProto struct

+ HostTransfers []*TPUHostTransferProto

+ func (*TPUHostTransferInfoProto) Descriptor() ([]byte, []int)

+ func (*TPUHostTransferInfoProto) ProtoMessage()

+ func (x *TPUHostTransferInfoProto) GetHostTransfers() []*TPUHostTransferProto

+ func (x *TPUHostTransferInfoProto) ProtoReflect() protoreflect.Message

+ func (x *TPUHostTransferInfoProto) Reset()

+ func (x *TPUHostTransferInfoProto) String() string

+ type TPUHostTransferProto struct

+ BufferOffset int64

+ Channel int64

+ Direction TPUHostTransferProto_TransferDirection

+ HostHandlerName string

+ IsLowerBits bool

+ Key string

+ OriginalType data.PrimitiveType

+ Shape *data.ShapeProto

+ func (*TPUHostTransferProto) Descriptor() ([]byte, []int)

+ func (*TPUHostTransferProto) ProtoMessage()

+ func (x *TPUHostTransferProto) GetBufferOffset() int64

+ func (x *TPUHostTransferProto) GetChannel() int64

+ func (x *TPUHostTransferProto) GetDirection() TPUHostTransferProto_TransferDirection

+ func (x *TPUHostTransferProto) GetHostHandlerName() string

+ func (x *TPUHostTransferProto) GetIsLowerBits() bool

+ func (x *TPUHostTransferProto) GetKey() string

+ func (x *TPUHostTransferProto) GetOriginalType() data.PrimitiveType

+ func (x *TPUHostTransferProto) GetShape() *data.ShapeProto

+ func (x *TPUHostTransferProto) ProtoReflect() protoreflect.Message

+ func (x *TPUHostTransferProto) Reset()

+ func (x *TPUHostTransferProto) String() string

+ type TPUHostTransferProto_TransferDirection int32

+ const TPUHostTransferProto_DEVICE_TO_HOST

+ const TPUHostTransferProto_HOST_TO_DEVICE

+ const TPUHostTransferProto_NONE

+ func (TPUHostTransferProto_TransferDirection) Descriptor() protoreflect.EnumDescriptor

+ func (TPUHostTransferProto_TransferDirection) EnumDescriptor() ([]byte, []int)

+ func (TPUHostTransferProto_TransferDirection) Type() protoreflect.EnumType

+ func (x TPUHostTransferProto_TransferDirection) Enum() *TPUHostTransferProto_TransferDirection

+ func (x TPUHostTransferProto_TransferDirection) Number() protoreflect.EnumNumber

+ func (x TPUHostTransferProto_TransferDirection) String() string

+ type TpuCompilationCacheServiceExternalClient interface

+ GetTpuProgram func(ctx context.Context, in *GetTpuProgramRequest, opts ...grpc.CallOption) (*GetTpuProgramResponseExternal, error)

+ func NewTpuCompilationCacheServiceExternalClient(cc grpc.ClientConnInterface) TpuCompilationCacheServiceExternalClient

+ type TpuCompilationCacheServiceExternalServer interface

+ GetTpuProgram func(context.Context, *GetTpuProgramRequest) (*GetTpuProgramResponseExternal, error)

+ type TpuCompilationRequestProto struct

+ ArgShapes []*framework.TensorShapeProto

+ FdefLib *framework.FunctionDefLibrary

+ Function *framework.NameAttrList

+ GraphDefVersion int32

+ GuaranteedConstants []*framework.TensorProto

+ Metadata *tpu.TPUCompileMetadataProto

+ MlirModule string

+ ReturnHloProtos bool

+ UnloadCacheOnSessionClose bool

+ UseExperimental bool

+ UseMlir bool

+ func (*TpuCompilationRequestProto) Descriptor() ([]byte, []int)

+ func (*TpuCompilationRequestProto) ProtoMessage()

+ func (x *TpuCompilationRequestProto) GetArgShapes() []*framework.TensorShapeProto

+ func (x *TpuCompilationRequestProto) GetFdefLib() *framework.FunctionDefLibrary

+ func (x *TpuCompilationRequestProto) GetFunction() *framework.NameAttrList

+ func (x *TpuCompilationRequestProto) GetGraphDefVersion() int32

+ func (x *TpuCompilationRequestProto) GetGuaranteedConstants() []*framework.TensorProto

+ func (x *TpuCompilationRequestProto) GetMetadata() *tpu.TPUCompileMetadataProto

+ func (x *TpuCompilationRequestProto) GetMlirModule() string

+ func (x *TpuCompilationRequestProto) GetReturnHloProtos() bool

+ func (x *TpuCompilationRequestProto) GetUnloadCacheOnSessionClose() bool

+ func (x *TpuCompilationRequestProto) GetUseExperimental() bool

+ func (x *TpuCompilationRequestProto) GetUseMlir() bool

+ func (x *TpuCompilationRequestProto) ProtoReflect() protoreflect.Message

+ func (x *TpuCompilationRequestProto) Reset()

+ func (x *TpuCompilationRequestProto) String() string

+ type TpuCompilationUidAndIndex struct

+ ProtoIndex int32

+ Uid int64

+ func (*TpuCompilationUidAndIndex) Descriptor() ([]byte, []int)

+ func (*TpuCompilationUidAndIndex) ProtoMessage()

+ func (x *TpuCompilationUidAndIndex) GetProtoIndex() int32

+ func (x *TpuCompilationUidAndIndex) GetUid() int64

+ func (x *TpuCompilationUidAndIndex) ProtoReflect() protoreflect.Message

+ func (x *TpuCompilationUidAndIndex) Reset()

+ func (x *TpuCompilationUidAndIndex) String() string

+ type UnimplementedTpuCompilationCacheServiceExternalServer struct

+ func (UnimplementedTpuCompilationCacheServiceExternalServer) GetTpuProgram(context.Context, *GetTpuProgramRequest) (*GetTpuProgramResponseExternal, error)

+ type UnsafeTpuCompilationCacheServiceExternalServer interface