Home
last modified time | relevance | path

Searched refs:sigmoid_op (Results 1 – 6 of 6) sorted by relevance

/external/XNNPACK/src/operators/
Dsigmoid-nc.c34 xnn_operator_t sigmoid_op = NULL; in xnn_create_sigmoid_nc_qu8() local
107 sigmoid_op = xnn_allocate_zero_simd_memory(sizeof(struct xnn_operator)); in xnn_create_sigmoid_nc_qu8()
108 if (sigmoid_op == NULL) { in xnn_create_sigmoid_nc_qu8()
115 sigmoid_op->lookup_table = xnn_allocate_simd_memory(256 * sizeof(uint8_t)); in xnn_create_sigmoid_nc_qu8()
116 if (sigmoid_op->lookup_table == NULL) { in xnn_create_sigmoid_nc_qu8()
123 uint8_t* lookup_table = sigmoid_op->lookup_table; in xnn_create_sigmoid_nc_qu8()
139 sigmoid_op->channels = channels; in xnn_create_sigmoid_nc_qu8()
140 sigmoid_op->input_pixel_stride = input_stride; in xnn_create_sigmoid_nc_qu8()
141 sigmoid_op->output_pixel_stride = output_stride; in xnn_create_sigmoid_nc_qu8()
143 sigmoid_op->type = xnn_operator_type_sigmoid_nc_qu8; in xnn_create_sigmoid_nc_qu8()
[all …]
Dunary-elementwise-nc.c733 xnn_operator_t sigmoid_op, in xnn_setup_sigmoid_nc_f32() argument
739 if (sigmoid_op->type != xnn_operator_type_sigmoid_nc_f32) { in xnn_setup_sigmoid_nc_f32()
742 xnn_operator_type_to_string(sigmoid_op->type)); in xnn_setup_sigmoid_nc_f32()
745 sigmoid_op->state = xnn_run_state_invalid; in xnn_setup_sigmoid_nc_f32()
748 sigmoid_op, in xnn_setup_sigmoid_nc_f32()
/external/XNNPACK/test/
Dsigmoid-operator-tester.h161 xnn_operator_t sigmoid_op = nullptr; in TestQU8() local
169 0, &sigmoid_op)); in TestQU8()
170 ASSERT_NE(nullptr, sigmoid_op); in TestQU8()
173 …std::unique_ptr<xnn_operator, decltype(&xnn_delete_operator)> auto_sigmoid_op(sigmoid_op, xnn_dele… in TestQU8()
177 sigmoid_op, in TestQU8()
183 xnn_run_operator(sigmoid_op, nullptr /* thread pool */)); in TestQU8()
218 xnn_operator_t sigmoid_op = nullptr; in TestF32() local
222 0, &sigmoid_op); in TestF32()
224 ASSERT_NE(nullptr, sigmoid_op); in TestF32()
227 …std::unique_ptr<xnn_operator, decltype(&xnn_delete_operator)> auto_sigmoid_op(sigmoid_op, xnn_dele… in TestF32()
[all …]
/external/XNNPACK/bench/
Dsigmoid.cc52 xnn_operator_t sigmoid_op = nullptr; in xnnpack_sigmoid_qu8() local
58 0 /* flags */, &sigmoid_op); in xnnpack_sigmoid_qu8()
59 if (status != xnn_status_success || sigmoid_op == nullptr) { in xnnpack_sigmoid_qu8()
65 sigmoid_op, in xnnpack_sigmoid_qu8()
75 status = xnn_run_operator(sigmoid_op, nullptr /* thread pool */); in xnnpack_sigmoid_qu8()
82 status = xnn_delete_operator(sigmoid_op); in xnnpack_sigmoid_qu8()
122 xnn_operator_t sigmoid_op = nullptr; in xnnpack_sigmoid_f32() local
125 0 /* flags */, &sigmoid_op); in xnnpack_sigmoid_f32()
126 if (status != xnn_status_success || sigmoid_op == nullptr) { in xnnpack_sigmoid_f32()
132 sigmoid_op, in xnnpack_sigmoid_f32()
[all …]
/external/tensorflow/tensorflow/core/kernels/rnn/
Dlstm_ops_gpu.cu.cc144 Eigen::internal::scalar_logistic_op<T> sigmoid_op; in lstm_gates() local
151 sigmoid_op(gates[0 * cell_size + gid] + b[0 * cell_size + act_id] + in lstm_gates()
155 sigmoid_op(gates[0 * cell_size + gid] + b[0 * cell_size + act_id]); in lstm_gates()
167 f_local = sigmoid_op(gates[f_offset + gid] + b[f_offset + act_id] + in lstm_gates()
170 f_local = sigmoid_op(gates[f_offset + gid] + b[f_offset + act_id] + in lstm_gates()
186 o_local = sigmoid_op(gates[3 * cell_size + gid] + in lstm_gates()
190 sigmoid_op(gates[3 * cell_size + gid] + b[3 * cell_size + act_id]); in lstm_gates()
/external/XNNPACK/include/
Dxnnpack.h1551 xnn_operator_t sigmoid_op,
2180 xnn_operator_t sigmoid_op,