Searched defs:TensorImpl (Results 1 – 9 of 9) sorted by relevance
/external/pytorch/c10/core/ |
D | TensorImpl.h | 414 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 …]
|
D | TensorImpl.cpp | 79 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/ |
D | exec_aten.h | 59 using TensorImpl = at::TensorImpl; variable 101 using TensorImpl = torch::executor::TensorImpl; variable
|
/external/pytorch/aten/src/ATen/templates/ |
D | TensorBody.h | 107 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/ |
D | tensor_impl.cpp | 43 TensorImpl::TensorImpl( in TensorImpl() function in executorch::runtime::etensor::TensorImpl
|
/external/pytorch/aten/src/ATen/ |
D | OpaqueTensorImpl.h | 30 : TensorImpl(key_set, data_type, device), in TensorImpl() function
|
/external/pytorch/torch/csrc/autograd/ |
D | custom_function.cpp | 30 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/ |
D | TensorImpl.java | 31 final class TensorImpl implements Tensor { class 475 private TensorImpl(long nativeHandle) { in TensorImpl() method in TensorImpl
|
/external/pytorch/aten/src/ATen/core/ |
D | TensorBase.h | 99 c10::intrusive_ptr<TensorImpl, UndefinedTensorImpl> tensor_impl) in TensorBase() 112 c10::intrusive_ptr<TensorImpl, UndefinedTensorImpl> tensor_impl) { in wrap_tensor_impl()
|