costs

type OpInfo struct {
	// The operation name.  There may be custom parameters in attrs.
	Op string `protobuf:"bytes,1,opt,name=op" json:"op,omitempty"`
	// Custom parameters impacting the behavior of the op.
	Attr   map[string]*tensorflow4.AttrValue `` /* 128-byte string literal not displayed */
	Inputs []*OpInfo_TensorProperties        `protobuf:"bytes,3,rep,name=inputs" json:"inputs,omitempty"`
	// Optional description of the op outputs
	Outputs []*OpInfo_TensorProperties `protobuf:"bytes,5,rep,name=outputs" json:"outputs,omitempty"`
	// Device on which the operation is run.
	Device *tensorflow5.DeviceProperties `protobuf:"bytes,4,opt,name=device" json:"device,omitempty"`
}

type OpInfo_TensorProperties struct {
	Dtype tensorflow2.DataType          `protobuf:"varint,1,opt,name=dtype,enum=tensorflow.DataType" json:"dtype,omitempty"`
	Shape *tensorflow1.TensorShapeProto `protobuf:"bytes,2,opt,name=shape" json:"shape,omitempty"`
	Value *tensorflow3.TensorProto      `protobuf:"bytes,3,opt,name=value" json:"value,omitempty"`
}

type OpPerformance struct {
	// The op
	Op *OpInfo `protobuf:"bytes,1,opt,name=op" json:"op,omitempty"`
	// The node name (optional). Makes it easier to associate the performance data
	// with a specific graph node.
	Node string `protobuf:"bytes,5,opt,name=node" json:"node,omitempty"`
	// Temporary memory used by this node (in bytes).
	TemporaryMemorySize int64 `protobuf:"varint,2,opt,name=temporary_memory_size,json=temporaryMemorySize" json:"temporary_memory_size,omitempty"`
	// Time it takes to run the op (in nanoseconds).
	ComputeCost int64 `protobuf:"varint,3,opt,name=compute_cost,json=computeCost" json:"compute_cost,omitempty"`
	// Analytical compute cost (in nanoseconds).
	ComputeTime int64 `protobuf:"varint,6,opt,name=compute_time,json=computeTime" json:"compute_time,omitempty"`
	// Analytical memory access cost (in nanoseconds).
	MemoryTime int64 `protobuf:"varint,7,opt,name=memory_time,json=memoryTime" json:"memory_time,omitempty"`
	// Percentage of theoretical compute performance.
	ComputeEfficiency float64 `protobuf:"fixed64,4,opt,name=compute_efficiency,json=computeEfficiency" json:"compute_efficiency,omitempty"`
	// Percentage of theoretical memory performance.
	MemoryEfficiency float64                 `protobuf:"fixed64,8,opt,name=memory_efficiency,json=memoryEfficiency" json:"memory_efficiency,omitempty"`
	OpMemory         *OpPerformance_OpMemory `protobuf:"bytes,9,opt,name=op_memory,json=opMemory" json:"op_memory,omitempty"`
}

type OpPerformanceList struct {
	OpPerformance []*OpPerformance `protobuf:"bytes,1,rep,name=op_performance,json=opPerformance" json:"op_performance,omitempty"`
}

type OpPerformance_OpMemory struct {
	// The output information may have memory usage and output shapes.
	OutputMemory []int64 `protobuf:"varint,1,rep,packed,name=output_memory,json=outputMemory" json:"output_memory,omitempty"`
	// Temporary memory allocated by this node.
	HostTempMemory   int64 `protobuf:"varint,2,opt,name=host_temp_memory,json=hostTempMemory" json:"host_temp_memory,omitempty"`
	DeviceTempMemory int64 `protobuf:"varint,3,opt,name=device_temp_memory,json=deviceTempMemory" json:"device_temp_memory,omitempty"`
	// The persisted_memory doesn't include outputs.
	HostPersistentMemory   int64 `protobuf:"varint,4,opt,name=host_persistent_memory,json=hostPersistentMemory" json:"host_persistent_memory,omitempty"`
	DevicePersistentMemory int64 `protobuf:"varint,5,opt,name=device_persistent_memory,json=devicePersistentMemory" json:"device_persistent_memory,omitempty"`
}