Home
last modified time | relevance | path

Searched defs:_inputs (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A Dpython_nccl.cpp132 PyObject *_inputs = nullptr, *_output = nullptr, *_streams = nullptr, in THCPModule_nccl_reduce() local
165 PyObject *_inputs = nullptr, *_outputs = nullptr, *_streams = nullptr, in THCPModule_nccl_all_reduce() local
198 PyObject *_inputs = nullptr, *_streams = nullptr, *_comms = nullptr; in THCPModule_nccl_broadcast() 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/tensorflow/tensorflow/python/data/ops/
H A Dmulti_device_iterator_ops.py156 def _inputs(self): member in _PerDeviceGenerator
198 def _inputs(self): member in _ReincarnatedPerDeviceGenerator
H A Ddataset_ops.py413 def _inputs(self): member in DatasetV2
4151 def _inputs(self): member in DatasetV1Adapter
4381 def _inputs(self): member in DatasetSource
4392 def _inputs(self): member in UnaryDataset
4416 def _inputs(self): member in _VariantDataset
4867 def _inputs(self): member in ZipDataset
4908 def _inputs(self): member in ConcatenateDataset
6426 def _inputs(self): member in _DirectedInterleaveDataset
H A Dreaders.py474 def _inputs(self): member in TFRecordDatasetV2
/aosp_15_r20/external/pytorch/torch/csrc/jit/backends/xnnpack/
H A Dxnnpack_graph_builder.h35 std::vector<torch::jit::Value*> _inputs; variable
/aosp_15_r20/external/federated-compute/fcp/tensorflow/
H A Dexternal_dataset.py53 def _inputs(self): member in ExternalDataset
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/coreml/
H A Dcoreml_executor.mm38 const std::vector<TensorData>* _inputs; field
/aosp_15_r20/external/tensorflow/tensorflow/lite/swift/Sources/
H A DSignatureRunner.swift94 private var _inputs: [String]? variable
/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_module_hooks.py514 def fw_pre_hook(module: nn.Module, _inputs): argument
517 def fw_hook(module: nn.Module, _inputs, _outputs): argument
520 def bw_hook(module: nn.Module, _inputs, _outputs): argument
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_hook.cpp165 const variable_list& _inputs /* grad_outputs */) -> variable_list { in operator ()()
/aosp_15_r20/out/soong/.intermediates/external/libtextclassifier/native/libtextclassifier_fbgen_lang_id_common_flatbuffers_model/gen/lang_id/common/flatbuffers/
Dmodel_generated.h415 …auto _inputs = _o->inputs.size() ? _fbb.CreateVector<flatbuffers::Offset<libtextclassifier3::saft_… in CreateModel() local
/aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/Llama-3.2-1B-Instruct/
H A Dtokenizer.json32069 "_inputs": 29657, number
/aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/Llama-3.2-3B-Instruct/
H A Dtokenizer.json32069 "_inputs": 29657, number
/aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/llama3-8B-instruct/
H A Dtokenizer.json32010 "_inputs": 29657, number