LostTech.TensorFlow : API Documentation

Type tfprof_logger

Namespace tensorflow.contrib.tfprof.tfprof_logger

Methods

Properties

Public static methods

void write_op_log(object graph, object log_dir, object op_log, object run_meta, bool add_trace)

Log provided 'op_log', and add additional model information below.

The API also assigns ops in tf.compat.v1.trainable_variables() an op type called '_trainable_variables'. The API also logs 'flops' statistics for ops with op.RegisterStatistics() defined. flops calculation depends on Tensor shapes defined in 'graph', which might not be complete. 'run_meta', if provided, completes the shape information with best effort.
Parameters
object graph
tf.Graph. If None and eager execution is not enabled, use default graph.
object log_dir
directory to write the log file.
object op_log
(Optional) OpLogProto proto to be written. If not provided, an new one is created.
object run_meta
(Optional) RunMetadata proto that helps flops computation using run time shape information.
bool add_trace
Whether to add python code trace information. Used to support "code" view.

object write_op_log_dyn(object graph, object log_dir, object op_log, object run_meta, ImplicitContainer<T> add_trace)

Log provided 'op_log', and add additional model information below.

The API also assigns ops in tf.compat.v1.trainable_variables() an op type called '_trainable_variables'. The API also logs 'flops' statistics for ops with op.RegisterStatistics() defined. flops calculation depends on Tensor shapes defined in 'graph', which might not be complete. 'run_meta', if provided, completes the shape information with best effort.
Parameters
object graph
tf.Graph. If None and eager execution is not enabled, use default graph.
object log_dir
directory to write the log file.
object op_log
(Optional) OpLogProto proto to be written. If not provided, an new one is created.
object run_meta
(Optional) RunMetadata proto that helps flops computation using run time shape information.
ImplicitContainer<T> add_trace
Whether to add python code trace information. Used to support "code" view.

Public properties

PythonFunctionContainer write_op_log_fn get;