Home
last modified time | relevance | path

Searched refs:n_input_channels (Results 1 – 16 of 16) sorted by relevance

/external/ComputeLibrary/src/core/NEON/kernels/convolution/common/
Dtensor.hpp92 int n_output_channels, n_rows, n_cols, n_input_channels; member
99 const int n_input_channels, in KernelShape()
104 n_input_channels(n_input_channels), in KernelShape()
113 return ((i*this->n_cols + j)*this->n_input_channels + ic)*this->n_output_channels + oc; in index()
117 return ((oc*this->n_input_channels + ic)*this->n_rows + i)*this->n_cols + j; in index()
123 return n_output_channels * n_rows * n_cols * n_input_channels; in size()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/
Ddepthfirst_driver.hpp75 virtual size_t get_working_size_per_thread(unsigned int n_input_channels) const = 0;
78 virtual void initialise_working_space(void *, unsigned int n_input_channels) const = 0;
148 unsigned int n_input_channels, in execute_internal() argument
168 …c_cast<uint8_t *>(working_space) + thread_id * this->get_working_size_per_thread(n_input_channels); in execute_internal()
169 this->initialise_working_space(thread_working_space, n_input_channels); in execute_internal()
175 const auto n_output_channels = n_input_channels * this->m_args.channel_multiplier; in execute_internal()
274 …size_t get_working_size(unsigned int n_threads, unsigned int n_input_channels) const override final in get_working_size()
276 return n_threads * this->get_working_size_per_thread(n_input_channels); in get_working_size()
Ddepthwise_depthfirst_generic.hpp235 size_t get_working_size_per_thread(const unsigned int n_input_channels) const override in get_working_size_per_thread()
238 args.input_channels = n_input_channels; in get_working_size_per_thread()
242 void initialise_working_space(void *buffer, unsigned int n_input_channels) const override in initialise_working_space()
245 args.input_channels = n_input_channels; in initialise_working_space()
Ddepthwise_depthfirst.hpp350 size_t get_working_size_per_thread(const unsigned int n_input_channels) const override in get_working_size_per_thread()
353 args.input_channels = n_input_channels; in get_working_size_per_thread()
359 void initialise_working_space(void *buffer, unsigned int n_input_channels) const override in initialise_working_space()
362 args.input_channels = n_input_channels; in initialise_working_space()
Ddepthwise_depthfirst_multiplier.hpp513 size_t get_working_size_per_thread(const unsigned int n_input_channels) const override in get_working_size_per_thread()
516 args.input_channels = n_input_channels; in get_working_size_per_thread()
520 void initialise_working_space(void *buffer, unsigned int n_input_channels) const override in initialise_working_space()
523 args.input_channels = n_input_channels; in initialise_working_space()
Ddepthwise_planar.hpp338 unsigned int n_input_channels, in execute_internal() argument
362 const auto n_output_channels = n_input_channels * this->m_args.channel_multiplier; in execute_internal()
/external/ComputeLibrary/src/core/NEON/kernels/assembly/
Dwinograd.hpp41 unsigned int n_input_channels; member
51 unsigned int n_input_channels, in ConvolutionArgs()
57 …: n_batches(n_batches), input_shape(input_shape), n_input_channels(n_input_channels), pad_top(pad_…
Ddepthwise_common.hpp83 … virtual size_t get_working_size(unsigned int n_threads, unsigned int n_input_channels) const = 0;
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/interleaves/
D8b_mla.cpp32 const unsigned int n_input_channels in generic_get_packed_size() argument
36 …return arm_gemm::roundup((long unsigned int) n_input_channels, per_iter) * kernel_rows * kernel_co… in generic_get_packed_size()
D8b_mla.hpp38 const unsigned int n_input_channels
/external/ComputeLibrary/src/core/NEON/kernels/convolution/winograd/
Dweight_transform.hpp68 start_ic < args.n_input_channels; in execute_internal()
72 const auto end_ic = std::min(args.n_input_channels, in execute_internal()
Dinput_transform.hpp127 args.n_input_channels, in execute_internal()
242 return sizeof(Workspace) + sizeof_inptr_array() + sizeof(TIn) * args.n_input_channels; in get_working_space_per_thread()
254 memset(buffer, 0, sizeof(TIn) * args.n_input_channels); in initialise_thread_working_space()
303 return sizeof(TIn) * input_points * args.n_input_channels; in get_working_space_per_thread()
Dwinograd_implementations.hpp310 conv_args.n_input_channels, // K in get_implementation()
324 ws.weight_ld_matrix = conv_args.n_input_channels * ws.weight_ld_row; in get_implementation()
327 ws.input_ld_row = iroundup(conv_args.n_input_channels, 4u); in get_implementation()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/pooling/
Ddepthfirst_driver.hpp67 virtual size_t get_working_size_per_thread(unsigned int n_input_channels) const = 0;
70 virtual void initialise_working_space(void *, unsigned int n_input_channels) const = 0;
/external/libxaac/encoder/
Dixheaace_mps_structure.h59 UWORD8 n_input_channels; member
Dixheaace_mps_enc.c176 const WORD32 num_ch_in = pstr_space_enc->n_input_channels; in ixheaace_mps_212_feed_de_inter_pre_scale()
339 pstr_space_info->num_total_input_channels = pstr_space_enc->n_input_channels; in ixheaace_mps_212_get_info()
407 num_ch_in = pstr_space_enc->n_input_channels; in ixheaace_mps_212_encode()
698 pstr_space_enc->n_input_channels = space_tree_description.num_out_channels; in ixheaace_mps_212_init()
708 for (ch = 0; ch < pstr_space_enc->n_input_channels; ch++) { in ixheaace_mps_212_init()
769 pstr_space_enc->n_frame_length * (pstr_space_enc->n_input_channels + num_ch_in_arb_dmx); in ixheaace_mps_212_init()