Searched refs:_reshaped_output (Results 1 – 4 of 4) sorted by relevance
93 _reshaped_output(), in NEFFTConvolutionLayer()203 …_reshaped_output.allocator()->init(_itransformed_output.info()->clone()->set_tensor_shape(reshaped… in configure()208 …const int end_right = _reshaped_output.info()->tensor_shape().x() - (kernel_size.x() - conv_info.… in configure()209 …const int end_botton = _reshaped_output.info()->tensor_shape().y() - (kernel_size.y() - conv_info.… in configure()219 …_extract_output_func.configure(&_reshaped_output, output_to_use, Coordinates(start_left, start_top… in configure()220 _reshaped_output.allocator()->allocate(); in configure()330 _reshaped_output.allocator()->import_memory(_itransformed_output.buffer()); in run()
96 _reshaped_output(), in CLFFTConvolutionLayer()212 …_reshaped_output.allocator()->init(_itransformed_output.info()->clone()->set_tensor_shape(reshaped… in configure()217 …const int end_right = _reshaped_output.info()->tensor_shape().x() - (kernel_size.x() - conv_info.… in configure()218 …const int end_botton = _reshaped_output.info()->tensor_shape().y() - (kernel_size.y() - conv_info.… in configure()228 …_extract_output_func.configure(compile_context, &_reshaped_output, output_to_use, Coordinates(star… in configure()336 _reshaped_output.allocator()->import_memory(_itransformed_output.cl_buffer()); in run()
155 Tensor _reshaped_output; variable
175 CLTensor _reshaped_output; variable