kernels

Documentation

Index

• Variables
• type TfCallbackData
  • func (*TfCallbackData) Descriptor() ([]byte, []int)
  • func (x *TfCallbackData) GetInputs() []*TfCallbackData_InputBufferDescription
  • func (x *TfCallbackData) GetOp() *framework.NodeDef
  • func (x *TfCallbackData) GetOutputs() []*TfCallbackData_OutputBufferDescription
  • func (*TfCallbackData) ProtoMessage()
  • func (x *TfCallbackData) ProtoReflect() protoreflect.Message
  • func (x *TfCallbackData) Reset()
  • func (x *TfCallbackData) String() string
• type TfCallbackData_BufferDescription
  • func (*TfCallbackData_BufferDescription) Descriptor() ([]byte, []int)
  • func (x *TfCallbackData_BufferDescription) GetShape() *framework.TensorShapeProto
  • func (x *TfCallbackData_BufferDescription) GetType() framework.DataType
  • func (*TfCallbackData_BufferDescription) ProtoMessage()
  • func (x *TfCallbackData_BufferDescription) ProtoReflect() protoreflect.Message
  • func (x *TfCallbackData_BufferDescription) Reset()
  • func (x *TfCallbackData_BufferDescription) String() string
• type TfCallbackData_InputBufferDescription
  • func (*TfCallbackData_InputBufferDescription) Descriptor() ([]byte, []int)
  • func (x *TfCallbackData_InputBufferDescription) GetBufferDescription() *TfCallbackData_BufferDescription
  • func (x *TfCallbackData_InputBufferDescription) GetValue() *framework.TensorProto
  • func (*TfCallbackData_InputBufferDescription) ProtoMessage()
  • func (x *TfCallbackData_InputBufferDescription) ProtoReflect() protoreflect.Message
  • func (x *TfCallbackData_InputBufferDescription) Reset()
  • func (x *TfCallbackData_InputBufferDescription) String() string
• type TfCallbackData_OutputBufferDescription
  • func (*TfCallbackData_OutputBufferDescription) Descriptor() ([]byte, []int)
  • func (x *TfCallbackData_OutputBufferDescription) GetBufferDescription() *TfCallbackData_BufferDescription
  • func (x *TfCallbackData_OutputBufferDescription) GetIsDynamicallyPadded() bool
  • func (*TfCallbackData_OutputBufferDescription) ProtoMessage()
  • func (x *TfCallbackData_OutputBufferDescription) ProtoReflect() protoreflect.Message
  • func (x *TfCallbackData_OutputBufferDescription) Reset()
  • func (x *TfCallbackData_OutputBufferDescription) String() string

Variables

var File_tensorflow_compiler_tf2xla_kernels_callback_proto protoreflect.FileDescriptor

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

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

Deprecated: Use TfCallbackData.ProtoReflect.Descriptor instead.

func (*TfCallbackData) GetInputs

func (x *TfCallbackData) GetInputs() []*TfCallbackData_InputBufferDescription

func (*TfCallbackData) GetOp

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

func (*TfCallbackData) GetOutputs

func (x *TfCallbackData) GetOutputs() []*TfCallbackData_OutputBufferDescription

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

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

Deprecated: Use TfCallbackData_BufferDescription.ProtoReflect.Descriptor instead.

func (*TfCallbackData_BufferDescription) GetShape

func (x *TfCallbackData_BufferDescription) GetShape() *framework.TensorShapeProto

func (*TfCallbackData_BufferDescription) GetType

func (x *TfCallbackData_BufferDescription) GetType() framework.DataType

func (*TfCallbackData_BufferDescription) ProtoMessage

func (*TfCallbackData_BufferDescription) ProtoMessage()

func (*TfCallbackData_BufferDescription) ProtoReflect

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

func (*TfCallbackData_BufferDescription) Reset

func (x *TfCallbackData_BufferDescription) Reset()

func (*TfCallbackData_BufferDescription) String

func (x *TfCallbackData_BufferDescription) String() 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

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

Deprecated: Use TfCallbackData_InputBufferDescription.ProtoReflect.Descriptor instead.

func (*TfCallbackData_InputBufferDescription) GetBufferDescription

func (x *TfCallbackData_InputBufferDescription) GetBufferDescription() *TfCallbackData_BufferDescription

func (*TfCallbackData_InputBufferDescription) GetValue

func (x *TfCallbackData_InputBufferDescription) GetValue() *framework.TensorProto

func (*TfCallbackData_InputBufferDescription) ProtoMessage

func (*TfCallbackData_InputBufferDescription) ProtoMessage()

func (*TfCallbackData_InputBufferDescription) ProtoReflect

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

func (*TfCallbackData_InputBufferDescription) Reset

func (x *TfCallbackData_InputBufferDescription) Reset()

func (*TfCallbackData_InputBufferDescription) String

func (x *TfCallbackData_InputBufferDescription) String() 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

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

Deprecated: Use TfCallbackData_OutputBufferDescription.ProtoReflect.Descriptor instead.

func (*TfCallbackData_OutputBufferDescription) GetBufferDescription

func (x *TfCallbackData_OutputBufferDescription) GetBufferDescription() *TfCallbackData_BufferDescription

func (*TfCallbackData_OutputBufferDescription) GetIsDynamicallyPadded

func (x *TfCallbackData_OutputBufferDescription) GetIsDynamicallyPadded() bool

func (*TfCallbackData_OutputBufferDescription) ProtoMessage

func (*TfCallbackData_OutputBufferDescription) ProtoMessage()

func (*TfCallbackData_OutputBufferDescription) ProtoReflect

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

func (*TfCallbackData_OutputBufferDescription) Reset

func (x *TfCallbackData_OutputBufferDescription) Reset()

func (*TfCallbackData_OutputBufferDescription) String

func (x *TfCallbackData_OutputBufferDescription) String() string