Lines Matching refs:output_vector
64 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()
106 ARM_COMPUTE_RETURN_ERROR_ON(num_slices > output_vector.size()); in validate()
113 …ARM_COMPUTE_RETURN_ON_ERROR(CLStridedSlice::validate(input, output_vector[k], slice_start, Coordin… in validate()