Home
last modified time | relevance | path

Searched defs:_tensor (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A DNEFillBorderKernel.h88 ITensor *_tensor; variable
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/utils/
H A DClAuxTensorHandler.h43 : _tensor() in _tensor() function
/aosp_15_r20/external/ComputeLibrary/src/cpu/utils/
H A DCpuAuxTensorHandler.h43 : _tensor() in _tensor() function
/aosp_15_r20/external/pytorch/functorch/dim/
H A Ddelayed_mul_tensor.py40 def _tensor(self): member in DelayedMulTensor
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/
H A DEdge.h148 Tensor *_tensor; variable
/aosp_15_r20/external/pytorch/torch/utils/benchmark/utils/
H A Dcompare.py9 from torch import tensor as _tensor unknown
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_engine.cpp255 PyObject* _tensor = PyTuple_GET_ITEM(tensors, i); in THPEngine_run_backward() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/
H A DOperationUtils.h120 Tensor _tensor; variable
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dsaver_test.py3138 def _tensor(): function
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupUCC.cpp40 ucc_datatype_t to_ucc_dType(at::Tensor _tensor) { in to_ucc_dType()
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_methods_invocations.py7121 def _tensor(shape, dtype=dtype, low=None, high=None): function
7149 def _tensor(shape, dtype=dtype, low=None, high=None): function
7204 def _tensor(shape, dtype=dtype, low=None, high=None): function
7930 def _tensor(shape, dtype=dtype, low=None, high=None): function
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DModule.cpp380 PyObject* _tensor = PyTuple_GET_ITEM(tensors_o, i); in THCPModule_cudaJiteratorCompileAndLaunchKernel() local
/aosp_15_r20/external/pytorch/test/
H A Dtest_reductions.py2916 def _assert_warning(_func, _tensor, _correction): argument
/aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/Llama-3.2-1B-Instruct/
H A Dtokenizer.json26477 "_tensor": 24065, number
/aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/Llama-3.2-3B-Instruct/
H A Dtokenizer.json26477 "_tensor": 24065, number
/aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/llama3-8B-instruct/
H A Dtokenizer.json26418 "_tensor": 24065, number