Visualizes the generated graph

Source: R/view.R

Visualizes the generated graph

view_function_graph(
  fn,
  args,
  ...,
  name = deparse(substitute(fn)),
  profiler = FALSE,
  concrete_fn = do.call(fn$get_concrete_fn, args),
  graph = concrete_fn$graph
)

Arguments

fn

TensorFlow function (returned from tf.function())
args

arguments passed to fun
...

other arguments passed to tensorflow::tensorboard()
name

string, provided to tensorboard
profiler

logical, passed on to tf.compat.v2.summary.trace_on() (only used in eager mode)
concrete_fn

a ConcreteFunction (only used in graph mode, ignored with a warning if executing eagerly)
graph

a tensorflow graph (only used in graph mode, ignored with a warning if executing eagerly)

Examples

if (FALSE) {
fn <- tf_function(function(x) autograph(if(x > 0) x * x else x))
view_function_graph(fn, list(tf$constant(5)))
}