Searched refs:elu_op (Results 1 – 5 of 5) sorted by relevance
| /external/XNNPACK/test/ |
| D | elu-operator-tester.h | 116 xnn_operator_t elu_op = nullptr; in TestF32() local
122 0, &elu_op)); in TestF32()
123 ASSERT_NE(nullptr, elu_op); in TestF32()
126 …std::unique_ptr<xnn_operator, decltype(&xnn_delete_operator)> auto_elu_op(elu_op, xnn_delete_opera… in TestF32()
130 elu_op, in TestF32()
136 xnn_run_operator(elu_op, nullptr /* thread pool */)); in TestF32()
|
| /external/XNNPACK/bench/ |
| D | elu.cc | 47 xnn_operator_t elu_op = nullptr; in xnnpack_elu_f32() local
50 1.0f /* alpha */, 0 /* flags */, &elu_op); in xnnpack_elu_f32()
51 if (status != xnn_status_success || elu_op == nullptr) { in xnnpack_elu_f32()
57 elu_op, in xnnpack_elu_f32()
67 status = xnn_run_operator(elu_op, nullptr /* thread pool */); in xnnpack_elu_f32()
74 status = xnn_delete_operator(elu_op); in xnnpack_elu_f32()
|
| /external/XNNPACK/src/operators/ |
| D | unary-elementwise-nc.c | 601 xnn_operator_t elu_op, in xnn_setup_elu_nc_f32() argument
607 if (elu_op->type != xnn_operator_type_elu_nc_f32) { in xnn_setup_elu_nc_f32()
610 xnn_operator_type_to_string(elu_op->type)); in xnn_setup_elu_nc_f32()
613 elu_op->state = xnn_run_state_invalid; in xnn_setup_elu_nc_f32()
616 elu_op, in xnn_setup_elu_nc_f32()
619 &elu_op->params.f32_elu, sizeof(elu_op->params.f32_elu)); in xnn_setup_elu_nc_f32()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 41 "elu_op.cc",
42 "elu_op.h",
|
| /external/XNNPACK/include/ |
| D | xnnpack.h | 1318 xnn_operator_t elu_op,
|