Home
last modified time | relevance | path

Searched refs:getTensorRef (Results 1 – 7 of 7) sorted by relevance

/external/pytorch/aten/src/ATen/core/
DTensor.h32 const Tensor& getTensorRef() const & { in getTensorRef() function
/external/pytorch/aten/src/ATen/native/
DLossNLL.cpp43 const Tensor& weight = weight_opt.getTensorRef(); in TORCH_META_FUNC()
113 const auto& weight = weight_opt.getTensorRef(); in TORCH_META_FUNC()
463 const Tensor& weight = weight_opt.getTensorRef(); in TORCH_IMPL_FUNC()
478 const Tensor& weight = weight_opt.getTensorRef(); in TORCH_IMPL_FUNC()
DNaiveConvolutionTranspose2d.cpp197 bias_opt.getTensorRef(), in TORCH_META_FUNC()
791 const Tensor& bias = bias_opt.getTensorRef(); in TORCH_IMPL_FUNC()
/external/pytorch/aten/src/ATen/native/cuda/
DLoss.cu600 const Tensor& weight = weight_opt.getTensorRef(); in TORCH_IMPL_FUNC()
614 const Tensor& weight = weight_opt.getTensorRef(); in TORCH_IMPL_FUNC()
DNaiveConvolutionTranspose2d.cu693 const Tensor& bias = bias_opt.getTensorRef(); in TORCH_IMPL_FUNC()
/external/pytorch/aten/src/ATen/native/mps/operations/
DLossOps.mm1100 const Tensor& weight = weight_opt.getTensorRef();
1115 const Tensor& weight = weight_opt.getTensorRef();
/external/pytorch/aten/src/ATen/
DTensorIterator.cpp82 return get()->getTensorRef(); in getTensor()