Home
last modified time | relevance | path

Searched refs:broadcast_value (Results 1 – 15 of 15) sorted by relevance

/external/ComputeLibrary/src/cpu/kernels/elementwise_binary/generic/neon/
Dimpl.h74 …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/
Dlist.h92 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()
Dqsymm16.cpp82 … 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/
Dqsymm16.cpp82 … 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()
Dimpl.cpp73 …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/
Dimpl.cpp73 …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/
Dimpl.cpp76 …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/
DCpuMulKernel.cpp198 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/
Dqsymm16.cpp82 … 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()
Dqasymm8_signed.cpp86 … 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()
Dqasymm8.cpp86 … 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/
Dimpl.h148 …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/
Dmul.h142 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/
DNELogicalKernel.cpp243 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/
Doptimized_ops.h1725 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()