Lines Matching refs:scratch_buffer
71 … ITensor *scratch_buffer, ITensor *output_state_out, ITensor *cell_state_out, ITensor *output, in configure() argument
79 scratch_buffer, output_state_out, cell_state_out, output); in configure()
93 … scratch_buffer->info(), output_state_out->info(), cell_state_out->info(), output->info(), in configure()
370 _concat_scratch_buffer.configure(scratch_inputs, scratch_buffer, Window::DimX); in configure()
382 …const ITensorInfo *scratch_buffer, const ITensorInfo *output_state_out, const ITensorInfo *cell_st… in validate() argument
390 scratch_buffer, output_state_out, cell_state_out, output); in validate()
399 … scratch_buffer, output_state_out, cell_state_out, output); in validate()
414 ARM_COMPUTE_RETURN_ERROR_ON(scratch_buffer->num_dimensions() > 2); in validate()
418 ARM_COMPUTE_RETURN_ERROR_ON(cell_bias->dimension(0) * 4 != scratch_buffer->dimension(0) in validate()
419 && cell_bias->dimension(0) * 3 != scratch_buffer->dimension(0)); in validate()
603 …ARM_COMPUTE_RETURN_ON_ERROR(NEConcatenateLayer::validate(inputs_vector_info_raw, scratch_buffer, W… in validate()