Searched refs:abs_function (Results 1 – 2 of 2) sorted by relevance
62 int32_t (*abs_function)(int32_t) = in main() local66 int32_t output_value = (*abs_function)(input_value); in main()
72 auto abs_function = [](float* output_ptr, const float* input_ptr, in abs_kernel()96 abs_function((float*)iter.data_ptr(0), (float*)iter.data_ptr(1), iter.numel()); in abs_kernel()110 abs_function((float*)output_operand.tensor_base().mutable_data_ptr(), in abs_kernel()