Home
last modified time | relevance | path

Searched refs:output_vector (Results 1 – 20 of 20) sorted by relevance

/external/ComputeLibrary/src/runtime/CL/functions/
DCLUnstack.cpp64 void CLUnstack::configure(const ICLTensor *input, const std::vector<ICLTensor *> &output_vector, in… in configure() argument
66 configure(CLKernelLibrary::get().get_compile_context(), input, output_vector, axis); in configure()
69 … &compile_context, const ICLTensor *input, const std::vector<ICLTensor *> &output_vector, int axis) in configure() argument
71 ARM_COMPUTE_LOG_PARAMS(input, output_vector, axis); in configure()
72 std::vector<ITensorInfo *> outputs_vector_info(output_vector.size()); in configure()
73 …std::transform(output_vector.begin(), output_vector.end(), outputs_vector_info.begin(), [](ICLTens… in configure()
94 …_strided_slice_vector[slice].configure(compile_context, input, output_vector[slice], slice_start, … in configure()
98 …tack::validate(const ITensorInfo *input, const std::vector<ITensorInfo *> &output_vector, int axis) in validate() argument
101 ARM_COMPUTE_RETURN_ERROR_ON(output_vector.empty()); in validate()
104 …const unsigned int num_slices = std::min(output_vector.size(), input->dimension(wrap_axis(axis, in… in validate()
[all …]
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEUnstack.cpp63 void NEUnstack::configure(const ITensor *input, const std::vector<ITensor *> &output_vector, int ax… in configure() argument
65 std::vector<ITensorInfo *> outputs_vector_info(output_vector.size()); in configure()
66 …std::transform(output_vector.begin(), output_vector.end(), outputs_vector_info.begin(), [](ITensor… in configure()
74 ARM_COMPUTE_LOG_PARAMS(input, output_vector, axis); in configure()
88 …_strided_slice_vector[slice].configure(input, output_vector[slice], slice_start, Coordinates(), Bi… in configure()
92 …tack::validate(const ITensorInfo *input, const std::vector<ITensorInfo *> &output_vector, int axis) in validate() argument
95 ARM_COMPUTE_RETURN_ERROR_ON(output_vector.empty()); in validate()
99 …const unsigned int num_slices = std::min(output_vector.size(), input->dimension(wrap_axis(axis, in… in validate()
101 ARM_COMPUTE_RETURN_ERROR_ON(num_slices > output_vector.size()); in validate()
109 …ARM_COMPUTE_RETURN_ON_ERROR(NEStridedSlice::validate(input, output_vector[k], slice_start, Coordin… in validate()
/external/ComputeLibrary/arm_compute/runtime/CL/functions/
DCLUnstack.h65 void configure(const ICLTensor *input, const std::vector<ICLTensor *> &output_vector, int axis);
75 …&compile_context, const ICLTensor *input, const std::vector<ICLTensor *> &output_vector, int axis);
84 …atus validate(const ITensorInfo *input, const std::vector<ITensorInfo *> &output_vector, int axis);
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dportable_tensor_utils.h294 void ReductionSumVector(const float* input_vector, float* output_vector, in ReductionSumVector() argument
296 PortableReductionSumVector(input_vector, output_vector, output_size, in ReductionSumVector()
300 void ReductionSumVector(const int32_t* input_vector, int32_t* output_vector, in ReductionSumVector() argument
302 PortableReductionSumVector(input_vector, output_vector, output_size, in ReductionSumVector()
306 void ReductionSumVector(const int8_t* input_vector, int32_t* output_vector, in ReductionSumVector() argument
308 PortableReductionSumVector(input_vector, output_vector, output_size, in ReductionSumVector()
312 void MeanStddevNormalization(const float* input_vector, float* output_vector, in MeanStddevNormalization() argument
314 PortableMeanStddevNormalization(input_vector, output_vector, v_size, n_batch); in MeanStddevNormalization()
Dportable_tensor_utils_impl.h214 OUTPUT* output_vector, int output_size, in PortableReductionSumVector() argument
221 output_vector[o] = result; in PortableReductionSumVector()
228 float* __restrict__ output_vector,
Dportable_tensor_utils.cc754 float* __restrict__ output_vector, in PortableMeanStddevNormalization() argument
772 output_vector[i] = (input_vector[i] - mean) * stddev_inv; in PortableMeanStddevNormalization()
775 output_vector += v_size; in PortableMeanStddevNormalization()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dneon_tensor_utils.h290 void ReductionSumVector(const float* input_vector, float* output_vector, in ReductionSumVector() argument
292 NEON_OR_PORTABLE(ReductionSumVector, input_vector, output_vector, output_size, in ReductionSumVector()
296 void ReductionSumVector(const int32_t* input_vector, int32_t* output_vector, in ReductionSumVector() argument
298 PortableReductionSumVector(input_vector, output_vector, output_size, in ReductionSumVector()
302 void ReductionSumVector(const int8_t* input_vector, int32_t* output_vector, in ReductionSumVector() argument
304 NEON_OR_PORTABLE(ReductionSumVector, input_vector, output_vector, output_size, in ReductionSumVector()
309 float* __restrict__ output_vector, int v_size, in MeanStddevNormalization() argument
311 NEON_OR_PORTABLE(MeanStddevNormalization, input_vector, output_vector, v_size, in MeanStddevNormalization()
Dsse_tensor_utils.h305 void ReductionSumVector(const float* input_vector, float* output_vector, in ReductionSumVector() argument
307 NEON_OR_PORTABLE(ReductionSumVector, input_vector, output_vector, output_size, in ReductionSumVector()
311 void ReductionSumVector(const int32_t* input_vector, int32_t* output_vector, in ReductionSumVector() argument
313 PortableReductionSumVector(input_vector, output_vector, output_size, in ReductionSumVector()
317 void ReductionSumVector(const int8_t* input_vector, int32_t* output_vector, in ReductionSumVector() argument
319 SSE_OR_PORTABLE(ReductionSumVector, input_vector, output_vector, output_size, in ReductionSumVector()
324 float* __restrict__ output_vector, int v_size, in MeanStddevNormalization() argument
326 PortableMeanStddevNormalization(input_vector, output_vector, v_size, n_batch); in MeanStddevNormalization()
Dneon_tensor_utils_impl.h176 void NeonReductionSumVector(const float* input_vector, float* output_vector,
179 void NeonReductionSumVector(const int8_t* input_vector, int32_t* output_vector,
188 float* __restrict__ output_vector, int v_size,
Dsse_tensor_utils_impl.h79 void SseReductionSumVector(const int8_t* input_vector, int32_t* output_vector,
Dneon_tensor_utils.cc2565 void NeonReductionSumVector(const float* input_vector, float* output_vector, in NeonReductionSumVector() argument
2584 output_vector[o] = sum; in NeonReductionSumVector()
2589 void NeonReductionSumVector(const int8_t* input_vector, int32_t* output_vector, in NeonReductionSumVector() argument
2611 output_vector[o] = sum; in NeonReductionSumVector()
2677 float* __restrict__ output_vector, int v_size, in NeonMeanStddevNormalization() argument
2772 vst1q_f32(output_vector + i + 0 * kFloatValuesPerNeonVector, in NeonMeanStddevNormalization()
2774 vst1q_f32(output_vector + i + 1 * kFloatValuesPerNeonVector, in NeonMeanStddevNormalization()
2776 vst1q_f32(output_vector + i + 2 * kFloatValuesPerNeonVector, in NeonMeanStddevNormalization()
2778 vst1q_f32(output_vector + i + 3 * kFloatValuesPerNeonVector, in NeonMeanStddevNormalization()
2782 output_vector[i] = (input_vector[i] - mean) * stddev_inv; in NeonMeanStddevNormalization()
[all …]
Dsse_tensor_utils.cc620 void SseReductionSumVector(const int8_t* input_vector, int32_t* output_vector, in SseReductionSumVector() argument
654 output_vector[row] = row_sum; in SseReductionSumVector()
/external/ComputeLibrary/arm_compute/runtime/NEON/functions/
DNEUnstack.h74 void configure(const ITensor *input, const std::vector<ITensor *> &output_vector, int axis);
83 …atus validate(const ITensorInfo *input, const std::vector<ITensorInfo *> &output_vector, int axis);
/external/tensorflow/tensorflow/lite/kernels/
Dwhile_test.cc161 std::vector<int> output_vector(1000009, 0); in TEST_F() local
162 output_vector[3] = 5; in TEST_F()
163 output_vector[4] = 7; in TEST_F()
164 CheckIntTensor(output2, {1000009}, output_vector); in TEST_F()
/external/armnn/src/backends/reference/workloads/
DLstmUtils.cpp41 armnn::Encoder<float>& output_vector, in MeanStddevNormalization() argument
66 output_vector.Set((input_vector.Get() - mean) * stddev_inv); in MeanStddevNormalization()
67 ++output_vector; in MeanStddevNormalization()
72 output_vector -= v_size * n_batch; in MeanStddevNormalization()
DLstmUtils.hpp24 armnn::Encoder<float>& output_vector,
/external/tensorflow/tensorflow/lite/kernels/internal/
Dportable_tensor_utils.h456 void ReductionSumVector(const float* input_vector, float* output_vector,
460 void ReductionSumVector(const int32_t* input_vector, int32_t* output_vector,
464 void ReductionSumVector(const int8_t* input_vector, int32_t* output_vector,
468 void MeanStddevNormalization(const float* input_vector, float* output_vector,
/external/iamf_tools/iamf/cli/proto_conversion/tests/
Dchannel_label_utils_test.cc233 std::vector<ChannelLabel::Label> output_vector = {kL2}; in TEST() local
235 ChannelLabelUtils::ConvertAndFillLabels(kInputLabels, output_vector), in TEST()
238 EXPECT_EQ(output_vector, kExpectedOutputVector); in TEST()
362 std::vector<ChannelLabel::Label> output_vector = {kL2}; in TEST() local
365 output_vector) in TEST()
/external/tensorflow/tensorflow/core/public/
DREADME.md76 auto output_vector = outputs[0].vec<float>();
77 if (output_vector(0) > 0.5) { ... }
/external/libopenapv/src/neon/
Doapv_tq_neon.c541 int16x8_t output_vector = vcombine_u16(vmovn_s32(coef_low_32b), vmovn_s32(coef_high_32b)); in oapv_quant_neon() local
544 output_vector = vbslq_s16(sign_mask, vnegq_s16(output_vector), output_vector); in oapv_quant_neon()
547 vst1q_s16(coef + i, output_vector); in oapv_quant_neon()