| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | SoftMaxKernel.cpp | 272 int64_t outer_stride = dim_size * inner_size; in _vec_softmax_backward() local
357 int64_t outer_stride = dim_size * inner_size; in _vec_softmax_backward() local
483 int64_t outer_stride = dim_size * inner_size; in _vec_log_softmax_backward() local
567 int64_t outer_stride = dim_size * inner_size; in _vec_log_softmax_backward() local
700 int64_t outer_stride = dim_size * dim_stride; in _vec_softmax() local
807 int64_t outer_stride = dim_size * dim_stride; in _vec_softmax() local
|
| H A D | SumKernel.cpp | 440 char * C10_RESTRICT data[2], int64_t outer_stride, int64_t out_stride, in vectorized_inner_sum()
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | meta_utils.py | 795 outer_stride, argument
825 t, outer_size, outer_stride, symbolic_context, callback, source argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | SoftMax.cpp | 177 int64_t outer_stride = dim_size * dim_stride; local
284 int64_t outer_stride = dim_size * dim_stride; in host_softmax_backward() local
|
| H A D | ReduceOps.cpp | 1784 const int64_t outer_stride = strides[1]; in std_var_all_cpu() local
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | custom_tensor.py | 37 def __tensor_unflatten__(inner_tensors, meta, outer_size, outer_stride): argument
|
| H A D | two_tensor.py | 47 def __tensor_unflatten__(inner_tensors, meta, outer_size, outer_stride): argument
|
| H A D | common_subclass.py | 246 def __tensor_unflatten__(cls, inner_tensors, meta, outer_size, outer_stride): argument
|
| /aosp_15_r20/external/executorch/kernels/optimized/cpu/ |
| H A D | op_log_softmax.cpp | 53 int64_t outer_stride = dim_size * dim_stride; in log_softmax_kernel() local
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_subclasses.py | 247 def __tensor_unflatten__(inner_tensors, metadata, outer_size, outer_stride): argument
297 def __tensor_unflatten__(inner_tensors, metadata, outer_size, outer_stride): argument
1222 def __tensor_unflatten__(inner_tensors, _, outer_size, outer_stride): argument
1560 def __tensor_unflatten__(inner_tensors, ctx, outer_size, outer_stride): argument
|
| H A D | test_autograd_function.py | 894 def __tensor_unflatten__(tensors, metadatas, outer_size, outer_stride): argument
|
| /aosp_15_r20/external/executorch/examples/models/llama/experimental/ |
| H A D | subclass.py | 197 cls, tensor_data_dict, attributes, outer_size=None, outer_stride=None argument
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | reduce_util.h | 223 const size_t outer_stride = size * stride; in apply_over_dim() local
|
| /aosp_15_r20/external/pytorch/torch/sparse/ |
| H A D | semi_structured.py | 181 outer_stride, argument
|
| /aosp_15_r20/external/pytorch/torch/nested/_internal/ |
| H A D | nested_tensor.py | 250 def __tensor_unflatten__(inner_tensors: Dict, meta, outer_size, outer_stride): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | SoftMax.cu | 261 const index_t outer_stride = inner_size * dim_size; in cunn_SpatialSoftMaxForward() local
319 const uint32_t outer_stride = inner_size * dim_size; in cunn_SpatialSoftMaxBackward() local
|
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | _python_dispatch.py | 400 def transform_subclass(t, callback, outer_size=None, outer_stride=None): argument
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/ |
| H A D | _api.py | 298 def __tensor_unflatten__(inner_tensors, flatten_spec, outer_size, outer_stride): argument
|
| /aosp_15_r20/external/pytorch/torch/distributed/ |
| H A D | _functional_collectives.py | 598 def __tensor_unflatten__(inner_tensors, meta, outer_size, outer_stride): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/ |
| H A D | NestedTensorTransformerFunctions.cu | 654 const int outer_stride = gridDim.x * blockDim.y; in __launch_bounds__() local
|