Home
last modified time | relevance | path

Searched defs:inner_size (Results 1 – 25 of 46) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DSoftMaxKernel.cpp269 int64_t inner_size, in _vec_softmax_backward()
353 int64_t inner_size, in _vec_softmax_backward()
480 int64_t inner_size, in _vec_log_softmax_backward()
563 int64_t inner_size, in _vec_log_softmax_backward()
695 int64_t inner_size, in _vec_softmax()
803 int64_t inner_size, in _vec_softmax()
897 int64_t inner_size, in _vec_logsoftmax()
1002 int64_t inner_size, in _vec_logsoftmax()
1131 int64_t inner_size = 1; in apply() local
1176 int64_t inner_size = 1; in apply() local
H A Dgroup_norm_kernel.cpp54 const int64_t inner_size = D * HxW; in GroupNormKernelImplInternal() local
97 const int64_t inner_size = D / K * K; in ColumnwiseMoments() local
132 const int64_t inner_size = D / K * K; in ColumnwiseMoments() local
572 const int64_t inner_size = HxW / K * K; in ComputeInternalGradients() local
725 const int64_t inner_size = D / K * K; in GammaBackward() local
773 const int64_t inner_size = D / K * K; in GammaBackward() local
825 const int64_t inner_size = C / K * K; in BetaBackward() local
850 const int64_t inner_size = C / K * K; in BetaBackward() local
949 const int64_t inner_size = C / K * K; in DsDbRowwiseMomentsChannelsLast() local
1253 const int64_t inner_size = D / K * K; in CalcInternalGradientsChannelsLast() local
[all …]
H A DCatKernel.cpp16 int64_t inner_size; member
H A DSerialStackImpl.h17 int64_t inner_size; member
H A DChannelShuffleKernel.cpp33 int64_t inner_size = image_size - (image_size % Vec::size()); in cpu_channel_shuffle() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DSoftMax.cu106 uint64_t outer_size, uint64_t inner_size) { in SpatialSoftMax_getGridSize()
121 uint64_t dim_size, uint64_t inner_size) { in SpatialSoftMax_getBlockSize()
137 uint64_t outer_size, uint64_t dim_size, uint64_t inner_size, in SpatialSoftMax_getLaunchSizes()
257 index_t outer_size, index_t dim_size, index_t inner_size) in cunn_SpatialSoftMaxForward()
315 uint32_t outer_size, uint32_t dim_size, uint32_t inner_size) in cunn_SpatialSoftMaxBackward()
833 int64_t inner_size = 1; in host_softmax() local
971 int64_t inner_size = 1; in host_softmax_backward() local
/aosp_15_r20/external/llvm-libc/src/__support/
H A Dblock.h155 LIBC_INLINE static size_t outer_size(size_t inner_size) { in outer_size()
162 LIBC_INLINE size_t inner_size() const { in inner_size() function
170 LIBC_INLINE static size_t inner_size(size_t outer_size) { in inner_size() function
/aosp_15_r20/external/pigweed/pw_allocator/bucket/public/pw_allocator/bucket/
H A Dsorted.h102 static constexpr auto MakeAddPredicate(size_t inner_size) { in MakeAddPredicate()
126 static constexpr auto MakeAddPredicate(size_t inner_size) { in MakeAddPredicate()
/aosp_15_r20/external/pigweed/pw_allocator/block/public/pw_allocator/block/
H A Dbasic.h226 size_t BasicBlock<Derived>::OuterSizeFromInnerSize(size_t inner_size) { in OuterSizeFromInnerSize()
239 size_t inner_size; in InnerSizeFromOuterSize() local
/aosp_15_r20/external/ruy/ruy/
H A Dtrace.h390 int inner_size = in str() local
403 int inner_size = layout.order == Order::kRowMajor ? layout.cols : layout.rows; in str() local
415 int inner_size = layout.order == Order::kRowMajor ? layout.cols : layout.rows; in str() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dfbgemm_utils.cpp63 const int64_t inner_size = D * H * W; in CopyToChannelsLast3dTensor() local
84 const int64_t inner_size = D * H * W; in CopyICFirst3dTensorToChannelsLast3dTensor() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DSoftMax.cpp169 int64_t inner_size = 1; local
276 int64_t inner_size = 1; in host_softmax_backward() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_c_api_client.cc476 int inner_size, xla::PjRtCApiClient* client) { in Convert2DCBuffersToCppBuffers()
525 size_t inner_size = numoutputs_args.num_outputs; in Execute() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DSelect.cpp64 const int inner_size = x.num_elements() / outer_size; in select() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dgather.h62 int inner_size = 1; in Gather() local
H A Darg_min_max.h54 int inner_size = 1; in ArgMinMax() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops_layout_helper.cc43 int inner_size) { in ShuffleArrayAttr()
/aosp_15_r20/external/python/cpython3/Modules/_blake2/clinic/
Dblake2b_impl.c.h41 int inner_size = 0; in py_blake2b_new() local
Dblake2s_impl.c.h41 int inner_size = 0; in py_blake2s_new() local
/aosp_15_r20/external/executorch/kernels/optimized/cpu/
H A Dop_log_softmax.cpp44 int64_t inner_size = 1; in log_softmax_kernel() local
/aosp_15_r20/external/pigweed/pw_allocator/
H A Dbuddy_allocator.cc114 size_t inner_size = bucket.max_inner_size(); in Allocate() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dgather_op.cc139 int64_t inner_size = 1; in Compute() local
/aosp_15_r20/external/pigweed/pw_allocator/public/pw_allocator/
H A Dblock_allocator_testing.h270 size_t inner_size = block->InnerSize() / alignment; in MeasureFragmentation() local
/aosp_15_r20/external/python/cpython3/Modules/_blake2/
Dblake2s_impl.c95 int inner_size, int last_node, int usedforsecurity) in py_blake2s_new_impl()
Dblake2b_impl.c95 int inner_size, int last_node, int usedforsecurity) in py_blake2b_new_impl()

12