Searched refs:clamp_op (Results 1 – 4 of 4) sorted by relevance
| /external/XNNPACK/src/ |
| D | clamp-nc.c | 31 xnn_operator_t clamp_op = NULL; in xnn_create_clamp_nc_u8() local
72 clamp_op = xnn_allocate_zero_simd_memory(sizeof(struct xnn_operator)); in xnn_create_clamp_nc_u8()
73 if (clamp_op == NULL) { in xnn_create_clamp_nc_u8()
78 clamp_op->channels = channels; in xnn_create_clamp_nc_u8()
79 clamp_op->input_pixel_stride = input_stride; in xnn_create_clamp_nc_u8()
80 clamp_op->output_pixel_stride = output_stride; in xnn_create_clamp_nc_u8()
81 clamp_op->u8_output_params = xnn_init_u8_output_params(output_min, output_max); in xnn_create_clamp_nc_u8()
83 clamp_op->type = xnn_operator_type_clamp_nc_u8; in xnn_create_clamp_nc_u8()
84 clamp_op->ukernel.type = xnn_ukernel_type_clamp; in xnn_create_clamp_nc_u8()
86 clamp_op->state = xnn_run_state_invalid; in xnn_create_clamp_nc_u8()
[all …]
|
| /external/XNNPACK/test/ |
| D | clamp-operator-tester.h | 127 xnn_operator_t clamp_op = nullptr; in TestU8() local
133 0, &clamp_op)); in TestU8()
134 ASSERT_NE(nullptr, clamp_op); in TestU8()
137 …std::unique_ptr<xnn_operator, decltype(&xnn_delete_operator)> auto_clamp_op(clamp_op, xnn_delete_o… in TestU8()
141 clamp_op, in TestU8()
147 xnn_run_operator(clamp_op, nullptr /* thread pool */)); in TestU8()
188 xnn_operator_t clamp_op = nullptr; in TestF32() local
194 0, &clamp_op)); in TestF32()
195 ASSERT_NE(nullptr, clamp_op); in TestF32()
198 …std::unique_ptr<xnn_operator, decltype(&xnn_delete_operator)> auto_clamp_op(clamp_op, xnn_delete_o… in TestF32()
[all …]
|
| /external/XNNPACK/include/ |
| D | xnnpack.h | 585 xnn_operator_t clamp_op,
1243 xnn_operator_t clamp_op,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_typed_visitor.h | 950 clamp_op = [](ElementwiseT low, ElementwiseT value, ElementwiseT high) {
957 std::move(ConvertTernaryFunction(clamp_op))));
967 clamp_op = [](ElementwiseT low, ElementwiseT value, ElementwiseT high) {
977 std::move(ConvertTernaryFunction(clamp_op))));
|