| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| 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()
|
| D | l2normalization.h | 35 const int outer_size = variable
61 const int outer_size = in L2Normalization() local
|
| 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
|
| D | concatenation.h | 51 int64_t outer_size = 1; in Concatenation() local
102 int64_t outer_size = 1; in ConcatenationWithScaling() local
|
| D | gather.h | 57 int outer_size = 1; in Gather() local
|
| D | arg_min_max.h | 48 int outer_size = 1; in ArgMinMax() local
|
| 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
|
| D | conv3d_transpose.h | 112 const int outer_size = in Conv3DTranspose() local
|
| D | select.h | 57 const int64_t outer_size = input_condition_shape.FlatSize(); in RankOneSelect() local
|
| D | lstm_cell.h | 279 const int outer_size = MatchingFlatSizeSkipDim( in LstmCell() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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
|
| 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()
|
| D | gather_op.cc | 138 int64_t outer_size = 1; in Compute() local
|
| /external/perfetto/src/base/ |
| 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
|
| /external/perfetto/src/profiling/memory/ |
| D | shared_ring_buffer.cc | 104 size_t outer_size = kMetaPageSize + size_ * 2 + kGuardSize; in ~SharedRingBuffer() local
154 size_t outer_size = kMetaPageSize + size * 2 + kGuardSize; in Initialize() local
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | Select.cpp | 63 const int outer_size = c.num_elements(); in select() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | l2normalization.h | 25 inline void L2Normalization(int32_t input_zero_point, int32_t outer_size, in L2Normalization()
|
| D | log_softmax.h | 27 int32_t diff_min, int32_t outer_size, int32_t depth, in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | l2norm.cc | 136 const int outer_size = in Eval() local
|
| D | roll.cc | 55 int stride = 1, outer_size, next_stride; in Pool() local
|
| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLSelectKernel.cpp | 105 …const size_t outer_size = x->info()->tensor_shape()[x->info()->tensor_shape().num_dimensions() - 1… in configure() local
|
| /external/ruy/example/ |
| D | parametrized_example.cc | 108 int outer_size = in run() local
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| 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
|
| /external/ComputeLibrary/src/cpu/kernels/select/generic/neon/ |
| D | impl.cpp | 127 const int outer_size = cond->info()->total_size() / cond->info()->element_size(); in select_op_not_same_rank() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | logsoftmax_quantized_test.cc | 230 const int outer_size = in RunOneLogSoftmaxTest() local
|