| /external/pytorch/aten/src/ATen/core/ |
| D | TensorAccessor.h | 73 class TensorAccessor : public TensorAccessorBase<T,N,PtrTraits,index_t> {
77 C10_HOST_DEVICE TensorAccessor( in TensorAccessor() function
83 C10_HOST_DEVICE TensorAccessor<T, N - 1, PtrTraits, index_t> operator[](index_t i) {
84 …return TensorAccessor<T,N-1,PtrTraits,index_t>(this->data_ + this->strides_[0]*i,this->sizes_+1,th…
87 C10_HOST_DEVICE const TensorAccessor<T, N-1, PtrTraits, index_t> operator[](index_t i) const {
88 …return TensorAccessor<T,N-1,PtrTraits,index_t>(this->data_ + this->strides_[0]*i,this->sizes_+1,th…
93 class TensorAccessor<T,1,PtrTraits,index_t> : public TensorAccessorBase<T,1,PtrTraits,index_t> {
97 C10_HOST_DEVICE TensorAccessor( in TensorAccessor() function
195 C10_DEVICE TensorAccessor<T, N - 1, PtrTraits, index_t> operator[](index_t i) {
198 …return TensorAccessor<T,N-1,PtrTraits,index_t>(this->data_ + this->strides_[0]*i, new_sizes, new_s…
[all …]
|
| D | TensorBase.h | 624 TensorAccessor<T,N> accessor() const& { in accessor()
633 return TensorAccessor<T,N>(ptr,sizes().data(),strides().data()); in accessor()
636 TensorAccessor<T,N> accessor() && = delete;
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | GridSamplerKernel.cpp | 475 ApplyGridSample(const TensorAccessor<const scalar_t, 4>& input) in ApplyGridSample()
540 inline void forward(TensorAccessor<scalar_t, 3>& out_slice, in forward()
541 const TensorAccessor<const scalar_t, 3>& inp_slice, in forward()
589 inline void backward(TensorAccessor<scalar_t, 3>* gInp_slice_ptr, in backward()
590 TensorAccessor<scalar_t, 3>& gGrid_slice, in backward()
591 const TensorAccessor<const scalar_t, 3>& gOut_slice, in backward()
592 const TensorAccessor<const scalar_t, 3>& inp_slice, in backward()
718 ApplyGridSample(const TensorAccessor<const scalar_t, 4>& input) in ApplyGridSample()
728 inline void forward(TensorAccessor<scalar_t, 3>& out_slice, in forward()
729 const TensorAccessor<const scalar_t, 3>& inp_slice, in forward()
[all …]
|
| D | HistogramKernel.cpp | 101 TensorAccessor<const input_t, 2> accessor_in = input.accessor<const input_t, 2>(); in histogramdd_cpu_contiguous()
107 … ? std::optional<TensorAccessor<const input_t, 1>>(weight.value().accessor<const input_t, 1>()) in histogramdd_cpu_contiguous()
108 : std::optional<TensorAccessor<const input_t, 1>>(); in histogramdd_cpu_contiguous()
|
| D | utils.h | 172 const TensorAccessor<index_t, 1>& crow_acc,
|
| /external/pytorch/aten/src/ATen/native/ |
| D | ConvolutionMM3d.cpp | 264 TensorAccessor<const scalar_t, 4> input, in slow_conv3d_update_output_frame()
265 TensorAccessor<scalar_t, 4> output, in slow_conv3d_update_output_frame()
266 TensorAccessor<const scalar_t, 2> weight, in slow_conv3d_update_output_frame()
268 TensorAccessor<const scalar_t, 2> finput, in slow_conv3d_update_output_frame()
313 TensorAccessor<scalar_t, 4> grad_input, in slow_conv3d_backward_update_grad_input_frame()
314 TensorAccessor<const scalar_t, 4> grad_output, in slow_conv3d_backward_update_grad_input_frame()
315 TensorAccessor<const scalar_t, 2> weight, in slow_conv3d_backward_update_grad_input_frame()
316 TensorAccessor<scalar_t, 2> fgrad_input, in slow_conv3d_backward_update_grad_input_frame()
466 TensorAccessor<scalar_t, 2> grad_weight, in slow_conv3d_backward_weight_frame()
467 TensorAccessor<const scalar_t, 4> grad_output, in slow_conv3d_backward_weight_frame()
[all …]
|
| D | batch_norm.h | 20 static TensorAccessor<scalar_t, 1> conditional_accessor_1d(const Tensor& t) { in conditional_accessor_1d()
22 return TensorAccessor<scalar_t, 1>(nullptr, nullptr, nullptr); in conditional_accessor_1d()
|
| D | TopKImpl.h | 33 TensorAccessor<scalar_t, 1> mode_values( in topk_impl_loop()
36 TensorAccessor<int64_t, 1> mode_indices( in topk_impl_loop()
39 TensorAccessor<const scalar_t, 1> tmp_values( in topk_impl_loop()
|
| D | ConvolutionMM2d.cpp | 223 TensorAccessor<const scalar_t, 3> input, in slow_conv2d_update_output_frame()
224 TensorAccessor<scalar_t, 3> output, in slow_conv2d_update_output_frame()
225 TensorAccessor<const scalar_t, 2> weight, in slow_conv2d_update_output_frame()
227 TensorAccessor<scalar_t, 2> finput, in slow_conv2d_update_output_frame()
287 TensorAccessor<scalar_t, 3> grad_input, in slow_conv2d_backward_update_grad_input_frame()
288 TensorAccessor<const scalar_t, 3> grad_output, in slow_conv2d_backward_update_grad_input_frame()
289 TensorAccessor<const scalar_t, 2> weight, in slow_conv2d_backward_update_grad_input_frame()
436 TensorAccessor<scalar_t, 2> grad_weight, in slow_conv2d_backward_weight_frame()
437 TensorAccessor<const scalar_t, 3> grad_output, in slow_conv2d_backward_weight_frame()
438 TensorAccessor<const scalar_t, 2> finput, in slow_conv2d_backward_weight_frame()
|
| D | Sorting.cpp | 127 TensorAccessor<scalar_t, 1> arr, in quick_select_template()
459 TensorAccessor<scalar_t, 1> tmp_values( in kthvalue_out_impl_cpu()
462 TensorAccessor<int64_t, 1> tmp_indices( in kthvalue_out_impl_cpu()
|
| /external/ComputeLibrary/arm_compute/core/ |
| D | WindowIterator.h | 53 class TensorAccessor
60 TensorAccessor(const ITensor &tensor) in TensorAccessor() function
|
| /external/pytorch/aten/src/ATen/native/sparse/ |
| D | SparseTensorMath.cpp | 1908 scalar_t binary_search_strided_rightmost(scalar_t search_val, TensorAccessor<scalar_t, 1>& sorted_a… in binary_search_strided_rightmost()
|