Home
last modified time | relevance | path

Searched defs:outer_size (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dl2normalization.h35 const int outer_size = variable
61 const int outer_size = in L2Normalization() local
Dsoftmax.h34 const int outer_size = in Softmax() local
87 const int outer_size = in Softmax() local
171 const int outer_size = in SoftmaxInt16() local
Dconcatenation.h49 int64_t outer_size = 1; in Concatenation() local
100 int64_t outer_size = 1; in ConcatenationWithScaling() local
Dgather.h40 int outer_size = 1; in Gather() local
Darg_min_max.h37 int outer_size = 1; in ArgMinMax() local
Dreference_ops.h325 int outer_size = 1; in Pack() local
352 int outer_size = 1; in Unpack() local
391 int outer_size = 1; in PackWithScaling() local
692 const int outer_size = MatchingFlatSizeSkipDim( in LstmCell() local
853 int64_t outer_size = 1; in Split() local
884 const int outer_size = in LocalResponseNormalization() local
909 const int outer_size = in LogSoftmax() local
959 const int outer_size = in LogSoftmax() local
1503 const int64_t outer_size = input_condition_shape.FlatSize(); in RankOneSelect() local
1698 int outer_size = 1; in Reverse() local
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dgather_functor_batched.h45 const SliceIndex outer_size = static_cast<SliceIndex>(params.dimension(1)); in HandleCopiesBatched() local
138 const int64 outer_size = params.dimension(1); in operator() local
Dgather_functor_batched_gpu.cu.h35 int64 outer_size, in GatherOpKernel()
Dgather_op.cc138 int64 outer_size = 1; in Compute() local
/external/perfetto/src/base/
Dpaged_memory.cc56 size_t outer_size = rounded_up_size + GuardSize() * 2; in Allocate() local
113 const size_t outer_size = RoundUpToSysPageSize(size_) + GuardSize() * 2; in ~PagedMemory() local
/external/perfetto/src/profiling/memory/
Dshared_ring_buffer.cc105 size_t outer_size = kMetaPageSize + size_ * 2 + kGuardSize; in ~SharedRingBuffer() local
155 size_t outer_size = kMetaPageSize + size * 2 + kGuardSize; in Initialize() local
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dsoftmax.cc50 const int outer_size = in SoftmaxQuantized() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dl2normalization.h23 inline void L2Normalization(int32_t input_zero_point, int32_t outer_size, in L2Normalization()
Dlog_softmax.h25 int32_t diff_min, int32_t outer_size, int32_t depth, in LogSoftmax()
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dsoftmax.cc69 const int outer_size = in SoftmaxHifimini() local
219 const int outer_size = in EvalHifi4() local
/external/tensorflow/tensorflow/lite/micro/kernels/
Dl2norm.cc125 const int outer_size = in Eval() local
Dunpack.cc46 int outer_size = 1; in UnpackImpl() local
Dpack.cc43 int outer_size = 1; in PackImpl() local
Dsplit_v.cc47 int64_t outer_size = 1; in SplitImpl() local
Dsplit.cc45 int64_t outer_size = 1; in SplitImpl() local
/external/tensorflow/tensorflow/lite/kernels/
Dl2norm.cc136 const int outer_size = in Eval() local
Dactivations.cc1227 const int outer_size = in LogSoftmaxEval() local
/external/ruy/example/
Dparametrized_example.cc108 int outer_size = in run() local
/external/tensorflow/tensorflow/lite/kernels/internal/
Dlogsoftmax_quantized_test.cc230 const int outer_size = in RunOneLogSoftmaxTest() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h1738 const int outer_size = variable
1767 const int outer_size = in L2Normalization() local
2945 const int outer_size = MatchingFlatSizeSkipDim( in LstmCell() local
4073 const int outer_size = in LogSoftmax() local

12