/external/ComputeLibrary/src/cpu/kernels/elementwise_binary/generic/neon/ |
D | impl.h | 74 …m_op_broadcast(const typename VectorType::type &a, const ScalarType &broadcast_value, const bool r… in elementwise_arithm_op_broadcast() argument 79 vec_type broadcast_vector = wrapper::vdup_n(broadcast_value, tag_type{}); in elementwise_arithm_op_broadcast() 121 …const InputScalarType broadcast_value = *reinterpret_cast<const InputScalarType *>(broadca… in elementwise_op() 123 …indow_start_x, window_end_x, window_step_x, non_broadcast_input_ptr, broadcast_value, output_ptr, … in elementwise_op() 127 …x) = (*scalar_func)(!is_broadcast_input_2 ? broadcast_value : a, !is_broadcast_input_2 ? a : broad… in elementwise_op() 255 …const ScalarType *non_broadcast_input_ptr, const ScalarType &broadcast_value, ScalarType *output_p… in elementwise_arithm_op_broadcast_loop() argument 261 …ptr + x, elementwise_arithm_op_broadcast<op, ScalarType, VectorType>(a, broadcast_value, reorder)); in elementwise_arithm_op_broadcast_loop() 341 …e_comp_op_broadcast(const InputVectorType &a, const InputScalarType &broadcast_value, const bool r… in elementwise_comp_op_broadcast() argument 343 …InputVectorType broadcast_vector = wrapper::vdup_n(broadcast_value, wrapper::traits::vector_128_ta… in elementwise_comp_op_broadcast() 349 …nst InputScalarType *non_broadcast_input_ptr, const InputScalarType &broadcast_value, uint8_t *out… in elementwise_comp_op_broadcast_8_loop() argument [all …]
|
/external/ComputeLibrary/src/cpu/kernels/sub/neon/ |
D | list.h | 92 const T broadcast_value = *reinterpret_cast<const T *>(broadcast_input.ptr()); in sub_same_neon() 93 const auto broadcast_value_vec = wrapper::vdup_n(broadcast_value, ExactTagType{}); in sub_same_neon() 112 … res = is_sat ? wrapper::sub_sat(broadcast_value, non_broadcast_v) : broadcast_v… in sub_same_neon()
|
D | qsymm16.cpp | 82 … const int16_t broadcast_value = *reinterpret_cast<const int16_t *>(broadcast_input.ptr()); in sub_qsymm16_neon() local 83 const int16x8_t broadcast_value_vec = vdupq_n_s16(broadcast_value); in sub_qsymm16_neon() 92 const float bfs = static_cast<int32_t>(broadcast_value) * broadcast_qinfo.scale; in sub_qsymm16_neon()
|
/external/ComputeLibrary/src/cpu/kernels/add/generic/neon/ |
D | qsymm16.cpp | 82 … const int16_t broadcast_value = *reinterpret_cast<const int16_t *>(broadcast_input.ptr()); in add_qsymm16_neon() local 83 const int16x8_t broadcast_value_vec = vdupq_n_s16(broadcast_value); in add_qsymm16_neon() 87 const float bfs = static_cast<int32_t>(broadcast_value) * broadcast_qinfo.scale; in add_qsymm16_neon()
|
D | impl.cpp | 73 …const ScalarType broadcast_value = *reinterpret_cast<const ScalarType *>(broadcast_input.ptr()… in add_same_neon() local 74 const auto broadcast_value_vec = wrapper::vdup_n(broadcast_value, ExactTagType{}); in add_same_neon() 89 …icy == ConvertPolicy::SATURATE) ? wrapper::add_sat(broadcast_value, non_broadcast_v) : broadcast_v… in add_same_neon() 432 const auto broadcast_value = *broadcast_input.ptr(); in add_sub_qasymm8_neon() local 433 const auto bf = vdupq_n_f32(float(broadcast_value) * scale2 + offset); in add_sub_qasymm8_neon() 434 const auto bfs = float(broadcast_value) * bf_scale + offset; in add_sub_qasymm8_neon() 615 const auto broadcast_value = *reinterpret_cast<const int8_t *>(broadcast_input.ptr()); in add_sub_qasymm8_signed_neon() local 616 const auto bf = vdupq_n_f32(float(broadcast_value) * scale2 + offset); in add_sub_qasymm8_signed_neon() 617 const auto bfs = float(broadcast_value) * bf_scale + offset; in add_sub_qasymm8_signed_neon()
|
/external/ComputeLibrary/src/cpu/kernels/elementwise_binary/generic/sve/ |
D | impl.cpp | 73 …const ScalarType broadcast_value = *reinterpret_cast<const ScalarType *>(broadcast_input.p… in elementwise_arithmetic_op() local 74 const auto broadcast_vector = svdup_n(broadcast_value); in elementwise_arithmetic_op() 179 …const InputScalarType broadcast_value = *reinterpret_cast<const InputScalarType *>(broadca… in elementwise_comparison_op() local 180 const auto broadcast_vector = svdup_n(broadcast_value); in elementwise_comparison_op()
|
/external/ComputeLibrary/src/cpu/kernels/add/generic/sve/ |
D | impl.cpp | 76 …const ScalarType broadcast_value = *reinterpret_cast<const ScalarType *>(broadcast_input.ptr()… in add_same_sve() local 77 const auto broadcast_value_vec = wrapper::svdup_n(broadcast_value); in add_same_sve()
|
/external/ComputeLibrary/src/cpu/kernels/ |
D | CpuMulKernel.cpp | 198 const auto broadcast_value = *reinterpret_cast<const T *>(broadcast_input.ptr()); in mul_saturate_quantized_8() local 199 const auto broadcast_value_vec = wrapper::vdup_n(broadcast_value, ExactTagType{}); in mul_saturate_quantized_8() 230 … const float tmp_in2 = Qasymm8QuantizationHelper<T>::dequantize(broadcast_value, broadcast_qinfo); in mul_saturate_quantized_8() 1082 … const int32_t broadcast_value = *reinterpret_cast<const int32_t *>(broadcast_input.ptr()); in mul_S32_S32_S32() local 1083 const auto broadcast_value_vec = vdupq_n_s32(broadcast_value); in mul_S32_S32_S32() 1112 …int64_t tmp = static_cast<int64_t>(broadcast_value) * static_cast<int64_t>(*(non_broadcast_input_p… in mul_S32_S32_S32() 1236 … const float broadcast_value = *reinterpret_cast<const float *>(broadcast_input.ptr()); in mul_F32_F32_F32() local 1237 const auto broadcast_value_vec = wrapper::vdup_n(broadcast_value, ExactTagType{}); in mul_F32_F32_F32() 1253 *(output_ptr + x) = broadcast_value * non_broadcast_v * scale; in mul_F32_F32_F32() 1336 const float broadcast_value = *reinterpret_cast<const float *>(broadcast_input.ptr()); in c_mul_F32_F32_F32_n() local [all …]
|
/external/ComputeLibrary/src/cpu/kernels/add/generic/sve2/ |
D | qsymm16.cpp | 82 … const int16_t broadcast_value = *reinterpret_cast<const int16_t *>(broadcast_input.ptr()); in add_qsymm16_sve2() local 83 const auto broadcast_value_vec = svdup_n_s16(broadcast_value); in add_qsymm16_sve2()
|
D | qasymm8_signed.cpp | 86 … const int8_t broadcast_value = *reinterpret_cast<const int8_t *>(broadcast_input.ptr()); in add_qasymm8_signed_sve2() local 87 const auto broadcast_value_vec = svdup_n_s8(broadcast_value); in add_qasymm8_signed_sve2()
|
D | qasymm8.cpp | 86 … const uint8_t broadcast_value = *reinterpret_cast<const uint8_t *>(broadcast_input.ptr()); in add_qasymm8_sve2() local 87 const svuint8_t broadcast_value_vec = svdup_n_u8(broadcast_value); in add_qasymm8_sve2()
|
/external/ComputeLibrary/src/cpu/kernels/elementwise_binary/generic/sve2/ |
D | impl.h | 148 …const ScalarType broadcast_value = *reinterpret_cast<const ScalarType *>(broadcast_input.p… in elementwise_arithmetic_quantized_op() 149 …st_value_f = Qasymm8QuantizationHelper<ScalarType>::dequantize(broadcast_value, broadcast_qi… in elementwise_arithmetic_quantized_op() 279 …const InputScalarType broadcast_value = *reinterpret_cast<const InputScalarType *>(broadca… in elementwise_comparison_quantized_op() 280 …lue_f = Qasymm8QuantizationHelper<InputScalarType>::dequantize(broadcast_value, broadcast_qi… in elementwise_comparison_quantized_op()
|
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
D | mul.h | 142 const int8 broadcast_value, in MulSimpleBroadcast() argument 145 const int16 input1_val = params.input1_offset + broadcast_value; in MulSimpleBroadcast()
|
/external/ComputeLibrary/src/core/NEON/kernels/ |
D | NELogicalKernel.cpp | 243 const uint8_t broadcast_value = *broadcast_in.ptr(); in run_binary() local 244 logical_func(non_broadcast_in.ptr(), broadcast_value, out.ptr(), len); in run_binary()
|
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
D | optimized_ops.h | 1725 float broadcast_value, const float* input2_data, in AddScalarBroadcast() argument 1733 const float32x4_t broadcast_value_dup = vdupq_n_f32(broadcast_value); in AddScalarBroadcast() 1748 auto x = broadcast_value + input2_data[i]; in AddScalarBroadcast() 2127 const uint8 broadcast_value, in MulSimpleBroadcast() argument 2129 const int16 input1_val = params.input1_offset + broadcast_value; in MulSimpleBroadcast() 2198 const float broadcast_value, in MulSimpleBroadcast() argument 2206 const float32x4_t broadcast_value_dup = vdupq_n_f32(broadcast_value); in MulSimpleBroadcast() 2221 float x = broadcast_value * input2_data[i]; in MulSimpleBroadcast()
|