tfprof

package
v0.0.0-...-d1695c4 Latest Latest
Warning

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

Go to latest
Published: Sep 29, 2021 License: Apache-2.0 Imports: 4 Imported by: 0

Documentation

Overview

Package tfprof is a generated protocol buffer package.

It is generated from these files:

tensorflow/tools/tfprof/tfprof_log.proto
tensorflow/tools/tfprof/tfprof_options.proto
tensorflow/tools/tfprof/tfprof_output.proto

It has these top-level messages:

CodeDef
OpLogEntry
OpLog
OptionsProto
TFProfTensorProto
TFGraphNodeProto
TFMultiGraphNodeProto

Index

Constants

View Source
const Default_OptionsProto_Step int64 = -1

Variables

This section is empty.

Functions

This section is empty.

Types

type CodeDef

type CodeDef struct {
	Traces           []*CodeDef_Trace `protobuf:"bytes,1,rep,name=traces" json:"traces,omitempty"`
	XXX_unrecognized []byte           `json:"-"`
}

It specifies the Python callstack that creates an op.

func (*CodeDef) Descriptor

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

func (*CodeDef) GetTraces

func (m *CodeDef) GetTraces() []*CodeDef_Trace

func (*CodeDef) ProtoMessage

func (*CodeDef) ProtoMessage()

func (*CodeDef) Reset

func (m *CodeDef) Reset()

func (*CodeDef) String

func (m *CodeDef) String() string

type CodeDef_Trace

type CodeDef_Trace struct {
	File             *string `protobuf:"bytes,1,opt,name=file" json:"file,omitempty"`
	Lineno           *int32  `protobuf:"varint,2,opt,name=lineno" json:"lineno,omitempty"`
	Function         *string `protobuf:"bytes,3,opt,name=function" json:"function,omitempty"`
	Line             *string `protobuf:"bytes,4,opt,name=line" json:"line,omitempty"`
	XXX_unrecognized []byte  `json:"-"`
}

func (*CodeDef_Trace) Descriptor

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

func (*CodeDef_Trace) GetFile

func (m *CodeDef_Trace) GetFile() string

func (*CodeDef_Trace) GetFunction

func (m *CodeDef_Trace) GetFunction() string

func (*CodeDef_Trace) GetLine

func (m *CodeDef_Trace) GetLine() string

func (*CodeDef_Trace) GetLineno

func (m *CodeDef_Trace) GetLineno() int32

func (*CodeDef_Trace) ProtoMessage

func (*CodeDef_Trace) ProtoMessage()

func (*CodeDef_Trace) Reset

func (m *CodeDef_Trace) Reset()

func (*CodeDef_Trace) String

func (m *CodeDef_Trace) String() string

type OpLog

type OpLog struct {
	LogEntries       []*OpLogEntry `protobuf:"bytes,1,rep,name=log_entries,json=logEntries" json:"log_entries,omitempty"`
	XXX_unrecognized []byte        `json:"-"`
}

func (*OpLog) Descriptor

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

func (*OpLog) GetLogEntries

func (m *OpLog) GetLogEntries() []*OpLogEntry

func (*OpLog) ProtoMessage

func (*OpLog) ProtoMessage()

func (*OpLog) Reset

func (m *OpLog) Reset()

func (*OpLog) String

func (m *OpLog) String() string

type OpLogEntry

type OpLogEntry struct {
	// op name.
	Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"`
	// float_ops is filled by tfprof Python API when called. It requires the
	// op has RegisterStatistics defined. Currently, Conv2D, MatMul, etc, are
	// implemented.
	FloatOps *int64 `protobuf:"varint,2,opt,name=float_ops,json=floatOps" json:"float_ops,omitempty"`
	// User can define extra op type information for an op. This allows the user
	// to select a group of ops precisely using op_type as a key.
	Types []string `protobuf:"bytes,3,rep,name=types" json:"types,omitempty"`
	// Used to support tfprof "code" view.
	CodeDef          *CodeDef `protobuf:"bytes,4,opt,name=code_def,json=codeDef" json:"code_def,omitempty"`
	XXX_unrecognized []byte   `json:"-"`
}

func (*OpLogEntry) Descriptor

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

func (*OpLogEntry) GetCodeDef

func (m *OpLogEntry) GetCodeDef() *CodeDef

func (*OpLogEntry) GetFloatOps

func (m *OpLogEntry) GetFloatOps() int64

func (*OpLogEntry) GetName

func (m *OpLogEntry) GetName() string

func (*OpLogEntry) GetTypes

func (m *OpLogEntry) GetTypes() []string

func (*OpLogEntry) ProtoMessage

func (*OpLogEntry) ProtoMessage()

func (*OpLogEntry) Reset

func (m *OpLogEntry) Reset()

func (*OpLogEntry) String

func (m *OpLogEntry) String() string

type OptionsProto

type OptionsProto struct {
	MaxDepth               *int64   `protobuf:"varint,1,opt,name=max_depth,json=maxDepth" json:"max_depth,omitempty"`
	MinBytes               *int64   `protobuf:"varint,2,opt,name=min_bytes,json=minBytes" json:"min_bytes,omitempty"`
	MinMicros              *int64   `protobuf:"varint,3,opt,name=min_micros,json=minMicros" json:"min_micros,omitempty"`
	MinParams              *int64   `protobuf:"varint,4,opt,name=min_params,json=minParams" json:"min_params,omitempty"`
	MinFloatOps            *int64   `protobuf:"varint,5,opt,name=min_float_ops,json=minFloatOps" json:"min_float_ops,omitempty"`
	MinOccurrence          *int64   `protobuf:"varint,17,opt,name=min_occurrence,json=minOccurrence" json:"min_occurrence,omitempty"`
	Step                   *int64   `protobuf:"varint,18,opt,name=step,def=-1" json:"step,omitempty"`
	OrderBy                *string  `protobuf:"bytes,7,opt,name=order_by,json=orderBy" json:"order_by,omitempty"`
	AccountTypeRegexes     []string `protobuf:"bytes,8,rep,name=account_type_regexes,json=accountTypeRegexes" json:"account_type_regexes,omitempty"`
	StartNameRegexes       []string `protobuf:"bytes,9,rep,name=start_name_regexes,json=startNameRegexes" json:"start_name_regexes,omitempty"`
	TrimNameRegexes        []string `protobuf:"bytes,10,rep,name=trim_name_regexes,json=trimNameRegexes" json:"trim_name_regexes,omitempty"`
	ShowNameRegexes        []string `protobuf:"bytes,11,rep,name=show_name_regexes,json=showNameRegexes" json:"show_name_regexes,omitempty"`
	HideNameRegexes        []string `protobuf:"bytes,12,rep,name=hide_name_regexes,json=hideNameRegexes" json:"hide_name_regexes,omitempty"`
	AccountDisplayedOpOnly *bool    `` /* 126-byte string literal not displayed */
	Select                 []string `protobuf:"bytes,14,rep,name=select" json:"select,omitempty"`
	Output                 *string  `protobuf:"bytes,15,opt,name=output" json:"output,omitempty"`
	DumpToFile             *string  `protobuf:"bytes,16,opt,name=dump_to_file,json=dumpToFile" json:"dump_to_file,omitempty"`
	XXX_unrecognized       []byte   `json:"-"`
}

Refers to tfprof_options.h/cc for documentation. Only used to pass tfprof options from Python to C++.

func (*OptionsProto) Descriptor

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

func (*OptionsProto) GetAccountDisplayedOpOnly

func (m *OptionsProto) GetAccountDisplayedOpOnly() bool

func (*OptionsProto) GetAccountTypeRegexes

func (m *OptionsProto) GetAccountTypeRegexes() []string

func (*OptionsProto) GetDumpToFile

func (m *OptionsProto) GetDumpToFile() string

func (*OptionsProto) GetHideNameRegexes

func (m *OptionsProto) GetHideNameRegexes() []string

func (*OptionsProto) GetMaxDepth

func (m *OptionsProto) GetMaxDepth() int64

func (*OptionsProto) GetMinBytes

func (m *OptionsProto) GetMinBytes() int64

func (*OptionsProto) GetMinFloatOps

func (m *OptionsProto) GetMinFloatOps() int64

func (*OptionsProto) GetMinMicros

func (m *OptionsProto) GetMinMicros() int64

func (*OptionsProto) GetMinOccurrence

func (m *OptionsProto) GetMinOccurrence() int64

func (*OptionsProto) GetMinParams

func (m *OptionsProto) GetMinParams() int64

func (*OptionsProto) GetOrderBy

func (m *OptionsProto) GetOrderBy() string

func (*OptionsProto) GetOutput

func (m *OptionsProto) GetOutput() string

func (*OptionsProto) GetSelect

func (m *OptionsProto) GetSelect() []string

func (*OptionsProto) GetShowNameRegexes

func (m *OptionsProto) GetShowNameRegexes() []string

func (*OptionsProto) GetStartNameRegexes

func (m *OptionsProto) GetStartNameRegexes() []string

func (*OptionsProto) GetStep

func (m *OptionsProto) GetStep() int64

func (*OptionsProto) GetTrimNameRegexes

func (m *OptionsProto) GetTrimNameRegexes() []string

func (*OptionsProto) ProtoMessage

func (*OptionsProto) ProtoMessage()

func (*OptionsProto) Reset

func (m *OptionsProto) Reset()

func (*OptionsProto) String

func (m *OptionsProto) String() string

type TFGraphNodeProto

type TFGraphNodeProto struct {
	// op name.
	Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"`
	// tensor value restored from checkpoint.
	TensorValue *TFProfTensorProto `protobuf:"bytes,15,opt,name=tensor_value,json=tensorValue" json:"tensor_value,omitempty"`
	// op execution time.
	ExecMicros            *int64 `protobuf:"varint,2,opt,name=exec_micros,json=execMicros" json:"exec_micros,omitempty"`
	AcceleratorExecMicros *int64 `protobuf:"varint,17,opt,name=accelerator_exec_micros,json=acceleratorExecMicros" json:"accelerator_exec_micros,omitempty"`
	CpuExecMicros         *int64 `protobuf:"varint,18,opt,name=cpu_exec_micros,json=cpuExecMicros" json:"cpu_exec_micros,omitempty"`
	// Total requested bytes by the op.
	RequestedBytes *int64 `protobuf:"varint,3,opt,name=requested_bytes,json=requestedBytes" json:"requested_bytes,omitempty"`
	// Number of parameters if available.
	Parameters *int64 `protobuf:"varint,4,opt,name=parameters" json:"parameters,omitempty"`
	// Number of float operations.
	FloatOps *int64 `protobuf:"varint,13,opt,name=float_ops,json=floatOps" json:"float_ops,omitempty"`
	// Device the op is assigned to.
	// Since an op can fire multiple kernel calls, there can be multiple devices.
	Devices []string `protobuf:"bytes,10,rep,name=devices" json:"devices,omitempty"`
	// The following are the aggregated stats from all accounted children and
	// the node itself. The actual children depend on the data structure used
	// (scope, graph).
	TotalExecMicros            *int64 `protobuf:"varint,6,opt,name=total_exec_micros,json=totalExecMicros" json:"total_exec_micros,omitempty"`
	TotalAcceleratorExecMicros *int64 `` /* 138-byte string literal not displayed */
	TotalCpuExecMicros         *int64 `protobuf:"varint,20,opt,name=total_cpu_exec_micros,json=totalCpuExecMicros" json:"total_cpu_exec_micros,omitempty"`
	TotalRequestedBytes        *int64 `protobuf:"varint,7,opt,name=total_requested_bytes,json=totalRequestedBytes" json:"total_requested_bytes,omitempty"`
	TotalParameters            *int64 `protobuf:"varint,8,opt,name=total_parameters,json=totalParameters" json:"total_parameters,omitempty"`
	TotalFloatOps              *int64 `protobuf:"varint,14,opt,name=total_float_ops,json=totalFloatOps" json:"total_float_ops,omitempty"`
	// shape information, if available.
	// TODO(xpan): Why is this repeated?
	Shapes      []*tensorflow.TensorShapeProto         `protobuf:"bytes,11,rep,name=shapes" json:"shapes,omitempty"`
	InputShapes map[int32]*tensorflow.TensorShapeProto `` /* 163-byte string literal not displayed */
	// Descendants of the graph. The actual descendants depend on the data
	// structure used (scope, graph).
	Children         []*TFGraphNodeProto `protobuf:"bytes,12,rep,name=children" json:"children,omitempty"`
	XXX_unrecognized []byte              `json:"-"`
}

A node in TensorFlow graph. Used by scope/graph view.

func (*TFGraphNodeProto) Descriptor

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

func (*TFGraphNodeProto) GetAcceleratorExecMicros

func (m *TFGraphNodeProto) GetAcceleratorExecMicros() int64

func (*TFGraphNodeProto) GetChildren

func (m *TFGraphNodeProto) GetChildren() []*TFGraphNodeProto

func (*TFGraphNodeProto) GetCpuExecMicros

func (m *TFGraphNodeProto) GetCpuExecMicros() int64

func (*TFGraphNodeProto) GetDevices

func (m *TFGraphNodeProto) GetDevices() []string

func (*TFGraphNodeProto) GetExecMicros

func (m *TFGraphNodeProto) GetExecMicros() int64

func (*TFGraphNodeProto) GetFloatOps

func (m *TFGraphNodeProto) GetFloatOps() int64

func (*TFGraphNodeProto) GetInputShapes

func (m *TFGraphNodeProto) GetInputShapes() map[int32]*tensorflow.TensorShapeProto

func (*TFGraphNodeProto) GetName

func (m *TFGraphNodeProto) GetName() string

func (*TFGraphNodeProto) GetParameters

func (m *TFGraphNodeProto) GetParameters() int64

func (*TFGraphNodeProto) GetRequestedBytes

func (m *TFGraphNodeProto) GetRequestedBytes() int64

func (*TFGraphNodeProto) GetShapes

func (m *TFGraphNodeProto) GetShapes() []*tensorflow.TensorShapeProto

func (*TFGraphNodeProto) GetTensorValue

func (m *TFGraphNodeProto) GetTensorValue() *TFProfTensorProto

func (*TFGraphNodeProto) GetTotalAcceleratorExecMicros

func (m *TFGraphNodeProto) GetTotalAcceleratorExecMicros() int64

func (*TFGraphNodeProto) GetTotalCpuExecMicros

func (m *TFGraphNodeProto) GetTotalCpuExecMicros() int64

func (*TFGraphNodeProto) GetTotalExecMicros

func (m *TFGraphNodeProto) GetTotalExecMicros() int64

func (*TFGraphNodeProto) GetTotalFloatOps

func (m *TFGraphNodeProto) GetTotalFloatOps() int64

func (*TFGraphNodeProto) GetTotalParameters

func (m *TFGraphNodeProto) GetTotalParameters() int64

func (*TFGraphNodeProto) GetTotalRequestedBytes

func (m *TFGraphNodeProto) GetTotalRequestedBytes() int64

func (*TFGraphNodeProto) ProtoMessage

func (*TFGraphNodeProto) ProtoMessage()

func (*TFGraphNodeProto) Reset

func (m *TFGraphNodeProto) Reset()

func (*TFGraphNodeProto) String

func (m *TFGraphNodeProto) String() string

type TFMultiGraphNodeProto

type TFMultiGraphNodeProto struct {
	// Name of the node.
	Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"`
	// code execution time.
	ExecMicros            *int64 `protobuf:"varint,2,opt,name=exec_micros,json=execMicros" json:"exec_micros,omitempty"`
	AcceleratorExecMicros *int64 `protobuf:"varint,12,opt,name=accelerator_exec_micros,json=acceleratorExecMicros" json:"accelerator_exec_micros,omitempty"`
	CpuExecMicros         *int64 `protobuf:"varint,13,opt,name=cpu_exec_micros,json=cpuExecMicros" json:"cpu_exec_micros,omitempty"`
	// Total requested bytes by the code.
	RequestedBytes *int64 `protobuf:"varint,3,opt,name=requested_bytes,json=requestedBytes" json:"requested_bytes,omitempty"`
	// Number of parameters if available.
	Parameters *int64 `protobuf:"varint,4,opt,name=parameters" json:"parameters,omitempty"`
	// Number of float operations.
	FloatOps *int64 `protobuf:"varint,5,opt,name=float_ops,json=floatOps" json:"float_ops,omitempty"`
	// The following are the aggregated stats from descendants.
	// The actual descendants depend on the data structure used.
	TotalExecMicros            *int64 `protobuf:"varint,6,opt,name=total_exec_micros,json=totalExecMicros" json:"total_exec_micros,omitempty"`
	TotalAcceleratorExecMicros *int64 `` /* 138-byte string literal not displayed */
	TotalCpuExecMicros         *int64 `protobuf:"varint,15,opt,name=total_cpu_exec_micros,json=totalCpuExecMicros" json:"total_cpu_exec_micros,omitempty"`
	TotalRequestedBytes        *int64 `protobuf:"varint,7,opt,name=total_requested_bytes,json=totalRequestedBytes" json:"total_requested_bytes,omitempty"`
	TotalParameters            *int64 `protobuf:"varint,8,opt,name=total_parameters,json=totalParameters" json:"total_parameters,omitempty"`
	TotalFloatOps              *int64 `protobuf:"varint,9,opt,name=total_float_ops,json=totalFloatOps" json:"total_float_ops,omitempty"`
	// TensorFlow graph nodes contained by the TFMultiGraphNodeProto.
	GraphNodes []*TFGraphNodeProto `protobuf:"bytes,10,rep,name=graph_nodes,json=graphNodes" json:"graph_nodes,omitempty"`
	// Descendants of the node. The actual descendants depend on the data
	// structure used.
	Children         []*TFMultiGraphNodeProto `protobuf:"bytes,11,rep,name=children" json:"children,omitempty"`
	XXX_unrecognized []byte                   `json:"-"`
}

A node that groups multiple TFGraphNodeProto. Depending on the 'view', the semantics of the TFmultiGraphNodeProto is different: code view: A node groups all TensorFlow graph nodes created by the

Python code.

op view: A node groups all TensorFlow graph nodes that are of type

of the op (e.g. MatMul, Conv2D).

func (*TFMultiGraphNodeProto) Descriptor

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

func (*TFMultiGraphNodeProto) GetAcceleratorExecMicros

func (m *TFMultiGraphNodeProto) GetAcceleratorExecMicros() int64

func (*TFMultiGraphNodeProto) GetChildren

func (m *TFMultiGraphNodeProto) GetChildren() []*TFMultiGraphNodeProto

func (*TFMultiGraphNodeProto) GetCpuExecMicros

func (m *TFMultiGraphNodeProto) GetCpuExecMicros() int64

func (*TFMultiGraphNodeProto) GetExecMicros

func (m *TFMultiGraphNodeProto) GetExecMicros() int64

func (*TFMultiGraphNodeProto) GetFloatOps

func (m *TFMultiGraphNodeProto) GetFloatOps() int64

func (*TFMultiGraphNodeProto) GetGraphNodes

func (m *TFMultiGraphNodeProto) GetGraphNodes() []*TFGraphNodeProto

func (*TFMultiGraphNodeProto) GetName

func (m *TFMultiGraphNodeProto) GetName() string

func (*TFMultiGraphNodeProto) GetParameters

func (m *TFMultiGraphNodeProto) GetParameters() int64

func (*TFMultiGraphNodeProto) GetRequestedBytes

func (m *TFMultiGraphNodeProto) GetRequestedBytes() int64

func (*TFMultiGraphNodeProto) GetTotalAcceleratorExecMicros

func (m *TFMultiGraphNodeProto) GetTotalAcceleratorExecMicros() int64

func (*TFMultiGraphNodeProto) GetTotalCpuExecMicros

func (m *TFMultiGraphNodeProto) GetTotalCpuExecMicros() int64

func (*TFMultiGraphNodeProto) GetTotalExecMicros

func (m *TFMultiGraphNodeProto) GetTotalExecMicros() int64

func (*TFMultiGraphNodeProto) GetTotalFloatOps

func (m *TFMultiGraphNodeProto) GetTotalFloatOps() int64

func (*TFMultiGraphNodeProto) GetTotalParameters

func (m *TFMultiGraphNodeProto) GetTotalParameters() int64

func (*TFMultiGraphNodeProto) GetTotalRequestedBytes

func (m *TFMultiGraphNodeProto) GetTotalRequestedBytes() int64

func (*TFMultiGraphNodeProto) ProtoMessage

func (*TFMultiGraphNodeProto) ProtoMessage()

func (*TFMultiGraphNodeProto) Reset

func (m *TFMultiGraphNodeProto) Reset()

func (*TFMultiGraphNodeProto) String

func (m *TFMultiGraphNodeProto) String() string

type TFProfTensorProto

type TFProfTensorProto struct {
	Dtype *tensorflow1.DataType `protobuf:"varint,1,opt,name=dtype,enum=tensorflow.DataType" json:"dtype,omitempty"`
	// Flatten tensor in row-major.
	// Only one of the following array is set.
	ValueDouble      []float64 `protobuf:"fixed64,2,rep,name=value_double,json=valueDouble" json:"value_double,omitempty"`
	ValueInt64       []int64   `protobuf:"varint,3,rep,name=value_int64,json=valueInt64" json:"value_int64,omitempty"`
	ValueStr         []string  `protobuf:"bytes,4,rep,name=value_str,json=valueStr" json:"value_str,omitempty"`
	XXX_unrecognized []byte    `json:"-"`
}

func (*TFProfTensorProto) Descriptor

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

func (*TFProfTensorProto) GetDtype

func (m *TFProfTensorProto) GetDtype() tensorflow1.DataType

func (*TFProfTensorProto) GetValueDouble

func (m *TFProfTensorProto) GetValueDouble() []float64

func (*TFProfTensorProto) GetValueInt64

func (m *TFProfTensorProto) GetValueInt64() []int64

func (*TFProfTensorProto) GetValueStr

func (m *TFProfTensorProto) GetValueStr() []string

func (*TFProfTensorProto) ProtoMessage

func (*TFProfTensorProto) ProtoMessage()

func (*TFProfTensorProto) Reset

func (m *TFProfTensorProto) Reset()

func (*TFProfTensorProto) String

func (m *TFProfTensorProto) String() string

Jump to

Keyboard shortcuts

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