| /external/tensorflow/tensorflow/python/framework/ |
| D | fast_tensor_util.pyx | 10 tensor_proto, np.ndarray[np.uint16_t, ndim=1] nparray): argument
22 tensor_proto, np.ndarray[np.uint16_t, ndim=1] nparray): argument
30 tensor_proto, np.ndarray[np.float32_t, ndim=1] nparray): argument
38 tensor_proto, np.ndarray[np.float64_t, ndim=1] nparray): argument
46 tensor_proto, np.ndarray[np.int32_t, ndim=1] nparray): argument
53 tensor_proto, np.ndarray[np.uint32_t, ndim=1] nparray): argument
60 tensor_proto, np.ndarray[np.int64_t, ndim=1] nparray): argument
67 tensor_proto, np.ndarray[np.uint64_t, ndim=1] nparray): argument
74 tensor_proto, np.ndarray[np.uint8_t, ndim=1] nparray): argument
82 tensor_proto, np.ndarray[np.uint16_t, ndim=1] nparray): argument
[all …]
|
| /external/executorch/backends/vulkan/runtime/graph/ops/impl/utils/ |
| D | TensorUtils.h | 69 T normalize(const T& nchw_dim, const int64_t ndim) { in normalize()
78 T nchw_dim_to_whcn_dim(const T& nchw_dim, const int64_t ndim) { in nchw_dim_to_whcn_dim()
|
| D | KernelUtils.cpp | 82 const int64_t ndim = in_sizes.size(); in calc_out_sizes_hw() local
125 const int64_t ndim = in_sizes.size(); in calc_transpose_out_sizes_hw() local
|
| /external/pytorch/torch/_numpy/ |
| D | _util.py | 47 def normalize_axis_index(ax, ndim, argname=None): argument
56 def normalize_axis_tuple(axis, ndim, argname=None, allow_duplicate=False): argument
120 def apply_keepdims(tensor, axis, ndim): argument
|
| /external/pytorch/aten/src/ATen/native/ |
| D | AdaptiveMaxPooling2d.cpp | 16 int ndim = input.ndimension(); in TORCH_META_FUNC() local
58 int64_t ndim = grad_output.ndimension(); in TORCH_META_FUNC() local
|
| D | im2col_shape_check.h | 47 int64_t ndim = input.ndimension(); in col2im_shape_check() local
177 int64_t ndim = input.ndimension(); in im2col_shape_check() local
|
| D | AdaptiveAveragePooling.cpp | 30 int64_t ndim = input.dim(); in adaptive_avg_pool2d_out_cpu_template() local
66 int64_t ndim = grad_output.ndimension(); in adaptive_avg_pool2d_backward_out_cpu_template() local
|
| D | Pool.h | 128 const int64_t ndim = input.ndimension(); in pool2d_shape_check() local
184 const int64_t ndim = input.ndimension(); in max_pool2d_backward_shape_check() local
214 const int64_t ndim = input.ndimension(); in avg_pool2d_backward_shape_check() local
299 const int64_t ndim = input.ndimension(); in max_pool3d_backward_shape_check() local
334 const int64_t ndim = input.ndimension(); in avg_pool3d_backward_shape_check() local
|
| D | ReduceOpsUtils.h | 114 inline DimVector make_dim_vector(OptionalIntArrayRef opt_dims, int64_t ndim) { in make_dim_vector()
170 inline Tensor review_reduce_result(const Tensor& result, int ndim, DimMask mask, bool keepdim) { in review_reduce_result()
199 int64_t ndim = self.dim(); in make_reduction() local
240 int64_t ndim = self.dim(); in make_reduction() local
405 int64_t ndim = self.dim(); in make_reduction() local
423 int64_t ndim = self.dim(); in make_reduction() local
|
| D | LinearAlgebraUtils.h | 351 const int64_t ndim = self.ndimension(); in _move_to_end() local
436 …e std::vector<int64_t> create_dim_backshift_permutation(int64_t dim0, int64_t dim1, int64_t ndim) { in create_dim_backshift_permutation()
457 int64_t ndim = permutation.size(); in create_reverse_permutation() local
584 auto ndim = input.dim(); in is_blas_compatible_column_major_order() local
605 auto ndim = input.dim(); in is_blas_compatible_row_major_order() local
|
| /external/pytorch/aten/src/ATen/ |
| D | TensorIteratorInternal.h | 29 const auto ndim = counter.size(); in get_data_ptrs() local
46 const auto ndim = shape.size(); in serial_for_each() local
|
| D | ExpandUtils.cpp | 21 auto ndim = dimsA > dimsB ? dimsA : dimsB; in infer_size_impl() local
66 int64_t ndim = static_cast<int64_t>(sizes.size()); in inferExpandGeometryImpl() local
153 size_t ndim = tensor_sizes.size(); in infer_dense_strides() local
|
| /external/pytorch/aten/src/ATen/native/mkldnn/xpu/detail/ |
| D | Conv.cpp | 19 int64_t ndim, in conv_dst_size()
52 const int64_t ndim, in conv_src_fmt()
70 const int64_t ndim, in conv_weight_fmt()
93 const int64_t ndim, in compatible_weight_dims()
132 auto ndim = src.ndimension(); in conv_get_md() local
|
| D | Deconv.cpp | 51 const int64_t ndim, in deconv_src_fmt()
73 const int64_t ndim, in deconv_weight_fmt()
92 int64_t ndim, in deconv_compatible_weight_dims()
129 auto ndim = src.ndimension(); in deconv_get_plain_md() local
|
| /external/executorch/backends/vulkan/runtime/graph/containers/ |
| D | Constant.cpp | 18 size_t ndim = t_sizes.size(); local
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | TensorModeKernel.cu | 153 int64_t ndim = ensure_nonempty_dim(self.dim()); in calculate_mode() local
188 int64_t ndim = ensure_nonempty_dim(self.dim()); in apply_mode() local
279 const TensorBase &self, int64_t dim, int64_t ndim) { in launch_apply_mode_kernel()
|
| D | SortImpl.cu | 9 int64_t ndim = self.dim(); in infer_dense_strides_dim_last() local
|
| D | TensorModeKernel.cpp | 23 int64_t ndim = ensure_nonempty_dim(self.dim()); in mode_kernel_impl() local
|
| /external/executorch/kernels/portable/cpu/ |
| D | op_constant_pad_nd.cpp | 32 size_t ndim, in apply_padding_to_dim()
112 size_t ndim = self.dim(); in constant_pad_nd_out_impl() local
|
| /external/pytorch/aten/src/ATen/cuda/ |
| D | CUDASparseDescriptors.cpp | 63 auto ndim = input.dim(); in createRawDnMatDescriptor() local
146 auto ndim = input.dim(); in CuSparseSpMatCsrDescriptor() local
|
| /external/pytorch/torch/csrc/inductor/aoti_runtime/ |
| D | arrayref_tensor.h | 166 int64_t ndim, in is_contiguous_strides_for_shape()
280 int64_t ndim, in reinterpret_tensor_wrapper()
293 int64_t ndim, in reinterpret_tensor_wrapper()
|
| /external/pytorch/torch/csrc/utils/ |
| D | tensor_numpy.cpp | 99 static std::vector<int64_t> to_aten_shape(int ndim, npy_intp* values) { in to_aten_shape()
109 int ndim = PySequence_Length(py_seq); in seq_to_aten_shape() local
234 int ndim = PyArray_NDIM(array); in tensor_from_numpy() local
|
| D | tensor_list.cpp | 21 int64_t ndim = static_cast<int64_t>(sizes.size()); in recursive_to_list() local
|
| /external/executorch/backends/vulkan/runtime/api/containers/ |
| D | Tensor.cpp | 25 const size_t ndim, in calculate_dim_order()
58 size_t ndim = sizes.size(); in calculate_strides() local
127 const size_t ndim = strides.size(); in unsqueeze_strides() local
144 int64_t ndim = sizes.size(); in calculate_padded_sizes() local
|
| /external/pytorch/aten/src/ATen/native/nested/ |
| D | NestedTensorUtils.cpp | 70 int64_t ndim = self.dim(); in chunk_nested_tensor() local
117 int64_t ndim = self.dim(); in split_with_sizes_nested() local
|