| /external/pytorch/torch/onnx/ |
| D | symbolic_opset13.py | 63 def split(g: jit_utils.GraphContext, self, split_size_or_sizes, dim, _outputs=None): argument
119 def split_with_sizes(g: jit_utils.GraphContext, self, split_sizes, dim, _outputs=None): argument
125 g: jit_utils.GraphContext, self, split_size_or_sizes, dim, _outputs=None argument
132 g: jit_utils.GraphContext, self, split_sizes, dim, _outputs=None argument
140 g: jit_utils.GraphContext, self, indices_or_sections, dim, _outputs=None argument
277 def unbind(g: jit_utils.GraphContext, self, dim=0, _outputs=None): argument
299 def nonzero_numpy(g: jit_utils.GraphContext, input, _outputs=None): argument
305 def where(g: jit_utils.GraphContext, condition, self=None, other=None, _outputs=None): argument
501 def unsafe_chunk(g: jit_utils.GraphContext, self, chunks, dim, _outputs=None): argument
|
| D | symbolic_opset11.py | 600 def split(g: jit_utils.GraphContext, self, split_size_or_sizes, dim, _outputs=None): argument
638 def split_with_sizes(g: jit_utils.GraphContext, self, split_sizes, dim, _outputs=None): argument
644 def unbind(g: jit_utils.GraphContext, self, dim=0, _outputs=None): argument
|
| D | symbolic_opset9.py | 1006 def unsafe_chunk(g: jit_utils.GraphContext, self, chunks, dim, _outputs=None): argument
1026 def split(g: jit_utils.GraphContext, self, split_size_or_sizes, dim, _outputs=None): argument
1053 g: jit_utils.GraphContext, self, split_size_or_sizes, dim, _outputs=None argument
1060 def split_with_sizes(g: jit_utils.GraphContext, self, split_sizes, dim, _outputs=None): argument
1070 g: jit_utils.GraphContext, self, split_sizes, dim, _outputs=None argument
1077 def unbind(g: jit_utils.GraphContext, self, dim=0, _outputs=None): argument
2193 def where(g: jit_utils.GraphContext, condition, self=None, other=None, _outputs=None): argument
4994 def nonzero_numpy(g: jit_utils.GraphContext, input, _outputs=None): argument
|
| D | symbolic_helper.py | 1187 def _unbind_helper(g: jit_utils.GraphContext, self, dim, _outputs): argument
1480 def _is_split_static(split_size_or_sizes, _outputs): argument
|
| /external/pytorch/torch/csrc/jit/backends/xnnpack/ |
| D | xnnpack_graph_builder.h | 36 std::vector<torch::jit::Value*> _outputs; variable
|
| /external/pytorch/torch/csrc/cuda/ |
| D | python_nccl.cpp | 165 PyObject *_inputs = nullptr, *_outputs = nullptr, *_streams = nullptr, in THCPModule_nccl_all_reduce() local
229 PyObject *_inputs = nullptr, *_outputs = nullptr, *_streams = nullptr, in THCPModule_nccl_all_gather() local
261 PyObject *_inputs = nullptr, *_outputs = nullptr, *_streams = nullptr, in THCPModule_nccl_reduce_scatter() local
|
| /external/ComputeLibrary/arm_compute/graph/ |
| D | INode.h | 262 std::vector<TensorID> _outputs; /**< Output of the node */ variable
|
| /external/tensorflow/tensorflow/lite/swift/Sources/ |
| D | SignatureRunner.swift | 97 private var _outputs: [String]? variable
|
| /external/pytorch/torch/csrc/autograd/ |
| D | python_hook.cpp | 164 const variable_list& _outputs, /* grad_inputs */ in operator ()()
|
| /external/pytorch/test/nn/ |
| D | test_module_hooks.py | 517 def fw_hook(module: nn.Module, _inputs, _outputs): argument
520 def bw_hook(module: nn.Module, _inputs, _outputs): argument
|