Home
last modified time | relevance | path

Searched refs:input_or_output (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/utils/tensorboard/
H A D_pytorch_graph.py92 def __init__(self, node_cpp, input_or_output=None): argument
106 if input_or_output:
107 self.input_or_output = input_or_output
194 node.input_or_output + "/" + node.debugName
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/
H A Dops.cc348 auto check_ctl_at_end = [&](TypeRange types, StringRef input_or_output) { in VerifyGenericTFGOperation() argument
356 op.emitOpError() << "found non-control " << input_or_output in VerifyGenericTFGOperation()
358 << " after control " << input_or_output in VerifyGenericTFGOperation()
/aosp_15_r20/external/pytorch/torch/_functorch/
H A Deager_transforms.py456 input_or_output = "inputs" if is_input else "outputs"