Home
last modified time | relevance | path

Searched refs:forget_gate_sm (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3073 auto forget_gate_sm = activ_temp_map.block(2 * output_depth, 0, output_depth, in LstmCell() local
3089 forget_gate_sm.unaryExpr(Eigen::internal::scalar_logistic_op<float>()) * in LstmCell()
Doptimized_ops.h2871 auto forget_gate_sm = activ_temp_map.block(2 * output_depth, 0, output_depth, in LstmCell() local
2887 forget_gate_sm.unaryExpr(Eigen::internal::scalar_logistic_op<float>()) * in LstmCell()