Searched defs:_tensor (Results 1 – 13 of 13) sorted by relevance
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NEFillBorderKernel.h | 88 ITensor *_tensor; variable
|
| /external/ComputeLibrary/src/gpu/cl/utils/ |
| D | ClAuxTensorHandler.h | 43 : _tensor() in _tensor() function
|
| /external/ComputeLibrary/src/cpu/utils/ |
| D | CpuAuxTensorHandler.h | 43 : _tensor() in _tensor() function
|
| /external/pytorch/functorch/dim/ |
| D | delayed_mul_tensor.py | 40 def _tensor(self): member in DelayedMulTensor
|
| /external/ComputeLibrary/arm_compute/graph/ |
| D | Edge.h | 148 Tensor *_tensor; variable
|
| /external/pytorch/torch/utils/benchmark/utils/ |
| D | compare.py | 9 from torch import tensor as _tensor unknown
|
| /external/pytorch/torch/csrc/autograd/ |
| D | python_engine.cpp | 255 PyObject* _tensor = PyTuple_GET_ITEM(tensors, i); in THPEngine_run_backward() local
|
| /external/pytorch/aten/src/ATen/native/mps/ |
| D | OperationUtils.h | 120 Tensor _tensor; variable
|
| /external/tensorflow/tensorflow/python/training/ |
| D | saver_test.py | 3138 def _tensor(): function
|
| /external/pytorch/torch/csrc/distributed/c10d/ |
| D | ProcessGroupUCC.cpp | 40 ucc_datatype_t to_ucc_dType(at::Tensor _tensor) { in to_ucc_dType()
|
| /external/pytorch/torch/csrc/cuda/ |
| D | Module.cpp | 380 PyObject* _tensor = PyTuple_GET_ITEM(tensors_o, i); in THCPModule_cudaJiteratorCompileAndLaunchKernel() local
|
| /external/pytorch/torch/testing/_internal/ |
| D | common_methods_invocations.py | 7121 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
|
| /external/pytorch/test/ |
| D | test_reductions.py | 2916 def _assert_warning(_func, _tensor, _correction): argument
|