Searched refs:_itransformed_output (Results 1 – 4 of 4) sorted by relevance
92 _itransformed_output(), in NEFFTConvolutionLayer()193 _memory_group.manage(&_itransformed_output); in configure()196 …_itransformed_output.allocator()->init(_output_reduced.info()->clone()->set_is_resizable(true).set… in configure()197 _itransform_output_func.configure(&_output_reduced, &_itransformed_output, itranform_info); in configure()201 TensorShape reshaped_shape = _itransformed_output.info()->tensor_shape(); in configure()203 …_reshaped_output.allocator()->init(_itransformed_output.info()->clone()->set_tensor_shape(reshaped… in configure()221 _itransformed_output.allocator()->allocate(); in configure()330 _reshaped_output.allocator()->import_memory(_itransformed_output.buffer()); in run()
95 _itransformed_output(), in CLFFTConvolutionLayer()202 _memory_group.manage(&_itransformed_output); in configure()205 …_itransformed_output.allocator()->init(_output_reduced.info()->clone()->set_is_resizable(true).set… in configure()206 …_itransform_output_func.configure(compile_context, &_output_reduced, &_itransformed_output, itranf… in configure()210 TensorShape reshaped_shape = _itransformed_output.info()->tensor_shape(); in configure()212 …_reshaped_output.allocator()->init(_itransformed_output.info()->clone()->set_tensor_shape(reshaped… in configure()229 _itransformed_output.allocator()->allocate(); in configure()336 _reshaped_output.allocator()->import_memory(_itransformed_output.cl_buffer()); in run()
154 Tensor _itransformed_output; variable
174 CLTensor _itransformed_output; variable