| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | SoftMaxKernel.cpp | 37 int64_t outer_size, in _vec_log_softmax_lastdim()
117 int64_t outer_size, in _vec_softmax_lastdim()
151 int64_t outer_size, in _vec_softmax_lastdim()
217 int64_t outer_size, in _vec_host_softmax_backward_lastdim()
268 int64_t outer_size, in _vec_softmax_backward()
352 int64_t outer_size, in _vec_softmax_backward()
479 int64_t outer_size, in _vec_log_softmax_backward()
562 int64_t outer_size, in _vec_log_softmax_backward()
673 int64_t outer_size = 1; in apply() local
694 int64_t outer_size, in _vec_softmax()
[all …]
|
| /aosp_15_r20/external/llvm-libc/src/__support/ |
| H A D | block.h | 153 LIBC_INLINE size_t outer_size() const { return next_ & SIZE_MASK; } in outer_size() function
155 LIBC_INLINE static size_t outer_size(size_t inner_size) { in outer_size() function
170 LIBC_INLINE static size_t inner_size(size_t outer_size) { in inner_size()
184 LIBC_INLINE static size_t inner_size_free(size_t outer_size) { in inner_size_free()
251 LIBC_INLINE constexpr Block(size_t outer_size) : next_(outer_size) { in Block()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | log_softmax.h | 32 const int outer_size = in LogSoftmax() local
84 const int outer_size = in LogSoftmax() local
155 const size_t outer_size, const size_t depth, in LogSoftmaxQuantized()
245 inline void LogSoftmax(const SoftmaxParams& params, const size_t outer_size, in LogSoftmax()
|
| H A D | softmax.h | 35 const int outer_size = in Softmax() local
88 const int outer_size = in Softmax() local
172 const int outer_size = in SoftmaxInt16() local
|
| H A D | reference_ops.h | 343 int outer_size = 1; in Pack() local
370 int outer_size = 1; in Unpack() local
409 int outer_size = 1; in PackWithScaling() local
478 int64_t outer_size = 1; in Split() local
509 const int outer_size = in LocalResponseNormalization() local
893 int outer_size = 1; in Reverse() local
921 int outer_size = 1; in ReverseSequence() local
|
| H A D | l2normalization.h | 35 const int outer_size = variable
61 const int outer_size = in L2Normalization() local
|
| H A D | concatenation.h | 51 int64_t outer_size = 1; in Concatenation() local
102 int64_t outer_size = 1; in ConcatenationWithScaling() local
|
| H A D | gather.h | 57 int outer_size = 1; in Gather() local
|
| H A D | arg_min_max.h | 48 int outer_size = 1; in ArgMinMax() local
|
| /aosp_15_r20/external/pigweed/pw_allocator/block/public/pw_allocator/block/ |
| H A D | testing.h | 79 size_t outer_size; member
107 size_t outer_size = in Preallocate() local
121 size_t outer_size = preallocation.outer_size; in Preallocate() local
|
| H A D | contiguous.h | 221 size_t outer_size = derived()->OuterSizeUnchecked(); in NextUnchecked() local
249 size_t outer_size = derived()->OuterSize() + next->OuterSize(); in DoMergeNext() local
|
| H A D | basic.h | 228 size_t outer_size; in OuterSizeFromInnerSize() local
237 size_t BasicBlock<Derived>::InnerSizeFromOuterSize(size_t outer_size) { in InnerSizeFromOuterSize()
|
| /aosp_15_r20/external/pigweed/pw_allocator/ |
| H A D | buddy_allocator.cc | 26 BuddyBlock::BuddyBlock(size_t outer_size) { in BuddyBlock()
115 size_t outer_size = BuddyBlock::OuterSizeFromInnerSize(inner_size); in Allocate() local
150 size_t outer_size = in Deallocate() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | SoftMax.cu | 106 uint64_t outer_size, uint64_t inner_size) { in SpatialSoftMax_getGridSize()
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()
829 int64_t outer_size = 1; in host_softmax() local
969 int64_t outer_size = 1; in host_softmax_backward() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | gather_functor_batched.h | 45 const SliceIndex outer_size = static_cast<SliceIndex>(params.dimension(1)); in HandleCopiesBatched() local
138 const int64_t outer_size = params.dimension(1); in operator() local
|
| H A D | gather_functor_batched_gpu.cu.h | 36 ValueOrVec* __restrict__ out, int64 outer_size, in GatherOpKernel()
124 const Index* indices, T* out, int64 outer_size, in LaunchGatherKernel()
|
| /aosp_15_r20/external/perfetto/src/base/ |
| H A D | paged_memory.cc | 57 size_t outer_size = rounded_up_size + GuardSize() * 2; in Allocate() local
114 const size_t outer_size = RoundUpToSysPageSize(size_) + GuardSize() * 2; in ~PagedMemory() local
|
| /aosp_15_r20/external/perfetto/src/profiling/memory/ |
| H A D | shared_ring_buffer.cc | 108 size_t outer_size = meta_page_size() + size_ * 2 + kGuardSize; in ~SharedRingBuffer() local
158 size_t outer_size = meta_page_size() + size * 2 + kGuardSize; in Initialize() local
|
| /aosp_15_r20/external/executorch/kernels/optimized/cpu/ |
| H A D | op_mul.cpp | 95 const size_t outer_size = getLeadingDims(out, out.dim() - 1); in handle_last_dim_broadcast() local
149 int64_t outer_size = 1; in handle_broadcast_mul() local
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | meta_utils.py | 794 outer_size, 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 | 167 int64_t outer_size = 1; local
274 int64_t outer_size = 1; in host_softmax_backward() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | pjrt_c_api_client.cc | 475 Convert2DCBuffersToCppBuffers(PJRT_Buffer*** c_lists, size_t outer_size, in Convert2DCBuffersToCppBuffers()
524 size_t outer_size = args.num_devices; in Execute() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/reference/ |
| H A D | Select.cpp | 63 const int outer_size = c.num_elements(); in select() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| H A D | l2normalization.h | 25 inline void L2Normalization(int32_t input_zero_point, int32_t outer_size, in L2Normalization()
|
| /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
|