Home
last modified time | relevance | path

Searched defs:sum_exp (Results 1 – 4 of 4) sorted by relevance

/external/XNNPACK/test/
Dsoftmax-operator-tester.h132 float sum_exp = 0.0f; in TestQU8() local
198 double sum_exp = 0.0; in TestF32() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsoftmax_op.cc72 auto sum_exp = XlaHelpers::ConvertElementType(reduce, type); in CrossEntropyWithLogits() local
/external/tensorflow/tensorflow/core/kernels/sparse/
Dkernels_gpu.cu.cc358 T sum_exp = 0; in CalculateRowSoftmax() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h3790 float sum_exp = 0.0f; in Softmax() local
3960 int32 sum_exp = 0; in SoftmaxInt8LUT() local
4143 float sum_exp = 0.0f; in LogSoftmax() local