/aosp_15_r20/external/pytorch/torch/onnx/ |
H A 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
|
H A 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
|
H A 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
|
H A 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
|
/aosp_15_r20/external/pytorch/torch/csrc/cuda/ |
H A 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
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/backends/xnnpack/ |
H A D | xnnpack_graph_builder.h | 36 std::vector<torch::jit::Value*> _outputs; variable
|
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/ |
H A D | INode.h | 262 std::vector<TensorID> _outputs; /**< Output of the node */ variable
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/swift/Sources/ |
H A D | SignatureRunner.swift | 97 private var _outputs: [String]? variable
|
/aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
H A D | python_hook.cpp | 164 const variable_list& _outputs, /* grad_inputs */ in operator ()()
|
/aosp_15_r20/external/pytorch/test/nn/ |
H A 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
|
/aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/Llama-3.2-1B-Instruct/ |
H A D | tokenizer.json | 38701 "_outputs": 36289, number
|
/aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/Llama-3.2-3B-Instruct/ |
H A D | tokenizer.json | 38701 "_outputs": 36289, number
|
/aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/llama3-8B-instruct/ |
H A D | tokenizer.json | 38642 "_outputs": 36289, number
|