Home
last modified time | relevance | path

Searched defs:dim_ (Results 1 – 25 of 36) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DSoftMax.cpp42 int64_t dim_ = maybe_wrap_dim(dim, input.dim()); in TORCH_META_FUNC() local
63 int64_t dim_ = maybe_wrap_dim(dim, input.dim()); in TORCH_META_FUNC() local
88 int64_t dim_ = maybe_wrap_dim(dim, grad.dim()); in TORCH_META_FUNC() local
121 int64_t dim_ = maybe_wrap_dim(dim, grad.dim()); in TORCH_META_FUNC() local
347 int64_t dim_ = maybe_wrap_dim(dim, input_.dim()); in TORCH_IMPL_FUNC() local
377 int64_t dim_ = maybe_wrap_dim(dim, input_.dim()); in TORCH_IMPL_FUNC() local
396 int64_t dim_ = maybe_wrap_dim(dim, grad.dim()); in TORCH_IMPL_FUNC() local
425 int64_t dim_ = maybe_wrap_dim(dim, grad.dim()); in TORCH_IMPL_FUNC() local
443 Tensor softmax(const Tensor& input_, const int64_t dim_, std::optional<ScalarType> dtype) { in softmax()
459 const int64_t dim_, in softmax_out()
[all …]
H A DSortingUtils.h21 int64_t dim_, in _reduction_with_indices_allocate_or_resize_output()
61 int64_t dim_, in _allocate_or_resize_output_with_indices()
H A DSorting.cpp419 int64_t dim_, in kthvalue_out_impl_cpu()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSoftMax.cpp537 const int64_t dim_, in softmax_sparse_cpu()
554 const int64_t dim_, in log_softmax_sparse_cpu()
572 int64_t dim_, in softmax_backward_sparse_cpu()
592 int64_t dim_, in log_softmax_backward_sparse_cpu()
609 Tensor _sparse_softmax(const Tensor& input_, const int64_t dim_, std::optional<ScalarType> dtype) { in _sparse_softmax()
627 Tensor _sparse_log_softmax(const Tensor& input_, const int64_t dim_, std::optional<ScalarType> dtyp… in _sparse_log_softmax()
H A DParamUtils.cpp18 const int64_t dim_, in softmax_sparse_input_preprocessing()
36 int64_t dim_, in softmax_backward_sparse_input_preprocessing()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/functions/
H A Dcomm.h30 int64_t dim_; member
42 int64_t dim_; member
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSoftMax.cu589 const int64_t dim_, in softmax_sparse_cuda()
606 const int64_t dim_, in log_softmax_sparse_cuda()
624 int64_t dim_, in softmax_backward_sparse_cuda()
644 int64_t dim_, in log_softmax_backward_sparse_cuda()
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesActivation.cpp22 const auto dim_ = maybe_wrap_dim(dim, rank) + 1; in glu_batch_rule() local
41 const auto dim_ = maybe_wrap_dim(dim, rank) + 1; in glu_backward_batch_rule() local
H A DBatchRulesLinearAlgebra.cpp416 const auto dim_ = getPhysicalDim(self_, true, dim); in cross_batch_rule() local
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DCPUApplyUtils.h60 int64_t dim_ = 0; member
89 int64_t dim_; member
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DSoftMax.cu820 Tensor host_softmax(const Tensor & input_, const int64_t dim_, const bool half_to_float, const Tens… in host_softmax()
958 void host_softmax_backward(const Tensor &grad_, const Tensor &output_, int64_t dim_, bool half_to_f… in host_softmax_backward()
1116 …ked_softmax_cuda(const Tensor& input_, const Tensor& mask_, const std::optional<int64_t> dim_, con… in masked_softmax_cuda()
1214 const std::optional<int64_t> dim_) { in masked_softmax_backward_cuda()
H A DSorting.cpp37 int64_t dim_, in kthvalue_out_impl_cuda()
/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Ddynamic_ir.h54 size_t dim_ = 0; variable
/aosp_15_r20/external/pytorch/torch/csrc/dynamo/
H A Dguards.h86 int64_t dim_; variable
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DSorting.cpp31 int64_t dim_, in quantized_topk_out_cpu()
/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/fuser/
H A Dpartition_desc.h53 size_t dim_; // dimension along which the chunk/concat occurs member
H A Dkernel_spec.h39 int64_t dim_; member
/aosp_15_r20/external/executorch/runtime/core/portable_type/
H A Dtensor_impl.h250 const ssize_t dim_; variable
/aosp_15_r20/external/executorch/examples/qualcomm/oss_scripts/llama2/runner/
H A Drunner.h267 int32_t dim_; variable
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dargument_spec.h59 unsigned dim_ : 8; member
/aosp_15_r20/external/pytorch/c10/core/
H A DTensorImpl.cpp929 const auto dim_ = sym_shape_meta.dim(); in empty_tensor_restride_symint() local
/aosp_15_r20/external/swiftshader/third_party/SPIRV-Tools/source/opt/
H A Dtypes.h340 spv::Dim dim_; variable
/aosp_15_r20/external/deqp-deps/SPIRV-Tools/source/opt/
Dtypes.h340 spv::Dim dim_; variable
/aosp_15_r20/external/angle/third_party/spirv-tools/src/source/opt/
H A Dtypes.h346 spv::Dim dim_; variable
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/operators/
H A Dmisc.cpp640 int64_t dim_ = std::get<int64_t>(argDim); in computeCat() local

12