Home
last modified time | relevance | path

Searched refs:scaled_input (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dsoftmax.cc145 float scaled_input = i * input->params.scale; in CalculateSoftmaxOpDataHifimini() local
147 std::exp((-scaled_input) * static_cast<float>(params->beta)); in CalculateSoftmaxOpDataHifimini()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dadd.h195 F0 scaled_input = F0::FromRaw(
197 F0 result = gemmlowp::SaturatingAdd(scaled_input, input_ready_scaled);
Dreference_ops.h292 F0 scaled_input = F0::FromRaw( in Sub16() local
294 F0 result = SaturatingSub(input_ready_scaled, scaled_input); in Sub16()
305 F0 scaled_input = F0::FromRaw( in Sub16() local
307 F0 result = SaturatingSub(scaled_input, input_ready_scaled); in Sub16()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc2661 auto scaled_input = rewriter.create<MulOp>(loc, operand, half); in matchAndRewrite() local
2662 auto tanh_op = rewriter.create<TanhOp>(loc, scaled_input); in matchAndRewrite()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h2129 F0 scaled_input = F0::FromRaw( in Add() local
2131 F0 result = gemmlowp::SaturatingAdd(scaled_input, input_ready_scaled); in Add()