Home
last modified time | relevance | path

Searched defs:TensorImpl (Results 1 – 9 of 9) sorted by relevance

/external/pytorch/c10/core/
DTensorImpl.h414 struct C10_API TensorImpl; variable
502 struct C10_API TensorImpl : public c10::intrusive_ptr_target { struct
516 TensorImpl( argument
538 TensorImpl( in TensorImpl() argument
546 TensorImpl( in TensorImpl() argument
565 TensorImpl& operator=(const TensorImpl&) = delete; argument
567 TensorImpl& operator=(TensorImpl&&) = delete; argument
592 // directly to on-TensorImpl fields. In particular, you never need to argument
607 IntArrayRef sizes() const { in sizes()
614 SymIntArrayRef sym_sizes() const { in sym_sizes()
[all …]
DTensorImpl.cpp79 TensorImpl::TensorImpl( in TensorImpl() function in c10::TensorImpl
108 TensorImpl::TensorImpl( in TensorImpl() function in c10::TensorImpl
127 TensorImpl::TensorImpl( in TensorImpl() function in c10::TensorImpl
134 TensorImpl::TensorImpl( in TensorImpl() function in c10::TensorImpl
/external/executorch/runtime/core/exec_aten/
Dexec_aten.h59 using TensorImpl = at::TensorImpl; variable
101 using TensorImpl = torch::executor::TensorImpl; variable
/external/pytorch/aten/src/ATen/templates/
DTensorBody.h107 c10::intrusive_ptr<TensorImpl, UndefinedTensorImpl> tensor_impl) in Tensor()
119 c10::intrusive_ptr<TensorImpl, UndefinedTensorImpl> tensor_impl) { in wrap_tensor_impl()
/external/executorch/runtime/core/portable_type/
Dtensor_impl.cpp43 TensorImpl::TensorImpl( in TensorImpl() function in executorch::runtime::etensor::TensorImpl
/external/pytorch/aten/src/ATen/
DOpaqueTensorImpl.h30 : TensorImpl(key_set, data_type, device), in TensorImpl() function
/external/pytorch/torch/csrc/autograd/
Dcustom_function.cpp30 std::unordered_map<at::TensorImpl*, size_t> inputs_mapping, in _process_forward_mode_AD()
257 const std::unordered_map<at::TensorImpl*, size_t>& inputs_mapping, in _process_backward_mode_ad()
/external/tensorflow/tensorflow/lite/java/src/main/java/org/tensorflow/lite/
DTensorImpl.java31 final class TensorImpl implements Tensor { class
475 private TensorImpl(long nativeHandle) { in TensorImpl() method in TensorImpl
/external/pytorch/aten/src/ATen/core/
DTensorBase.h99 c10::intrusive_ptr<TensorImpl, UndefinedTensorImpl> tensor_impl) in TensorBase()
112 c10::intrusive_ptr<TensorImpl, UndefinedTensorImpl> tensor_impl) { in wrap_tensor_impl()