Home
last modified time | relevance | path

Searched defs:_outputs (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset13.py63 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 Dsymbolic_opset11.py600 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 Dsymbolic_opset9.py1006 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 Dsymbolic_helper.py1187 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 Dpython_nccl.cpp165 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 Dxnnpack_graph_builder.h36 std::vector<torch::jit::Value*> _outputs; variable
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/
H A DINode.h262 std::vector<TensorID> _outputs; /**< Output of the node */ variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/swift/Sources/
H A DSignatureRunner.swift97 private var _outputs: [String]? variable
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_hook.cpp164 const variable_list& _outputs, /* grad_inputs */ in operator ()()
/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_module_hooks.py517 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 Dtokenizer.json38701 "_outputs": 36289, number
/aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/Llama-3.2-3B-Instruct/
H A Dtokenizer.json38701 "_outputs": 36289, number
/aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/llama3-8B-instruct/
H A Dtokenizer.json38642 "_outputs": 36289, number