kernels

package
v2.12.0 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Dec 8, 2023 License: BSD-3-Clause Imports: 5 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

View Source
var File_tensorflow_compiler_tf2xla_kernels_callback_proto protoreflect.FileDescriptor

Functions

This section is empty.

Types

type TfCallbackData

type TfCallbackData struct {
	Op      *framework.NodeDef                        `protobuf:"bytes,1,opt,name=op" json:"op,omitempty"`
	Inputs  []*TfCallbackData_InputBufferDescription  `protobuf:"bytes,2,rep,name=inputs" json:"inputs,omitempty"`
	Outputs []*TfCallbackData_OutputBufferDescription `protobuf:"bytes,3,rep,name=outputs" json:"outputs,omitempty"`
	// contains filtered or unexported fields
}

func (*TfCallbackData) Descriptor deprecated

func (*TfCallbackData) Descriptor() ([]byte, []int)

Deprecated: Use TfCallbackData.ProtoReflect.Descriptor instead.

func (*TfCallbackData) GetInputs

func (*TfCallbackData) GetOp

func (x *TfCallbackData) GetOp() *framework.NodeDef

func (*TfCallbackData) GetOutputs

func (*TfCallbackData) ProtoMessage

func (*TfCallbackData) ProtoMessage()

func (*TfCallbackData) ProtoReflect

func (x *TfCallbackData) ProtoReflect() protoreflect.Message

func (*TfCallbackData) Reset

func (x *TfCallbackData) Reset()

func (*TfCallbackData) String

func (x *TfCallbackData) String() string

type TfCallbackData_BufferDescription

type TfCallbackData_BufferDescription struct {
	Shape *framework.TensorShapeProto `protobuf:"bytes,1,opt,name=shape" json:"shape,omitempty"`
	Type  *framework.DataType         `protobuf:"varint,2,opt,name=type,enum=tensorflow.DataType" json:"type,omitempty"`
	// contains filtered or unexported fields
}

func (*TfCallbackData_BufferDescription) Descriptor deprecated

func (*TfCallbackData_BufferDescription) Descriptor() ([]byte, []int)

Deprecated: Use TfCallbackData_BufferDescription.ProtoReflect.Descriptor instead.

func (*TfCallbackData_BufferDescription) GetShape

func (*TfCallbackData_BufferDescription) GetType

func (*TfCallbackData_BufferDescription) ProtoMessage

func (*TfCallbackData_BufferDescription) ProtoMessage()

func (*TfCallbackData_BufferDescription) ProtoReflect

func (*TfCallbackData_BufferDescription) Reset

func (*TfCallbackData_BufferDescription) String

type TfCallbackData_InputBufferDescription

type TfCallbackData_InputBufferDescription struct {
	BufferDescription *TfCallbackData_BufferDescription `protobuf:"bytes,1,opt,name=buffer_description,json=bufferDescription" json:"buffer_description,omitempty"`
	// The input value might be already fixed at the compilation time.
	// This value may or may not be present.
	Value *framework.TensorProto `protobuf:"bytes,2,opt,name=value" json:"value,omitempty"`
	// contains filtered or unexported fields
}

func (*TfCallbackData_InputBufferDescription) Descriptor deprecated

func (*TfCallbackData_InputBufferDescription) Descriptor() ([]byte, []int)

Deprecated: Use TfCallbackData_InputBufferDescription.ProtoReflect.Descriptor instead.

func (*TfCallbackData_InputBufferDescription) GetBufferDescription

func (*TfCallbackData_InputBufferDescription) GetValue

func (*TfCallbackData_InputBufferDescription) ProtoMessage

func (*TfCallbackData_InputBufferDescription) ProtoMessage()

func (*TfCallbackData_InputBufferDescription) ProtoReflect

func (*TfCallbackData_InputBufferDescription) Reset

func (*TfCallbackData_InputBufferDescription) String

type TfCallbackData_OutputBufferDescription

type TfCallbackData_OutputBufferDescription struct {
	BufferDescription *TfCallbackData_BufferDescription `protobuf:"bytes,1,opt,name=buffer_description,json=bufferDescription" json:"buffer_description,omitempty"`
	// Whether the buffer stores dynamically padded data: in that case, actual
	// concrete dimensions need to be stored after the buffer.
	IsDynamicallyPadded *bool `protobuf:"varint,2,opt,name=is_dynamically_padded,json=isDynamicallyPadded" json:"is_dynamically_padded,omitempty"`
	// contains filtered or unexported fields
}

func (*TfCallbackData_OutputBufferDescription) Descriptor deprecated

func (*TfCallbackData_OutputBufferDescription) Descriptor() ([]byte, []int)

Deprecated: Use TfCallbackData_OutputBufferDescription.ProtoReflect.Descriptor instead.

func (*TfCallbackData_OutputBufferDescription) GetBufferDescription

func (*TfCallbackData_OutputBufferDescription) GetIsDynamicallyPadded

func (x *TfCallbackData_OutputBufferDescription) GetIsDynamicallyPadded() bool

func (*TfCallbackData_OutputBufferDescription) ProtoMessage

func (*TfCallbackData_OutputBufferDescription) ProtoReflect

func (*TfCallbackData_OutputBufferDescription) Reset

func (*TfCallbackData_OutputBufferDescription) String

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL