Searched refs:scratch_inputs (Results 1 – 3 of 3) sorted by relevance
518 std::vector<SimpleTensor<T>> scratch_inputs; in compute_reference() local521 scratch_inputs.emplace_back(std::move(input_gate)); in compute_reference()523 scratch_inputs.emplace_back(std::move(cell_state_out)); in compute_reference()524 scratch_inputs.emplace_back(std::move(forget_gate)); in compute_reference()525 scratch_inputs.emplace_back(std::move(output)); in compute_reference()526 scratch = reference::concatenate_layer(scratch_inputs, scratch, Window::DimX); in compute_reference()
360 std::vector<const ITensor *> scratch_inputs; in configure() local363 scratch_inputs.emplace_back(input_gate_out); in configure()365 scratch_inputs.emplace_back(&_cell_state_out1); in configure()366 scratch_inputs.emplace_back(forget_gate_out); in configure()367 scratch_inputs.emplace_back(output_gate_out); in configure()368 _concat_scratch_buffer.configure(scratch_inputs, scratch_buffer, Window::DimX); in configure()
385 std::vector<const ICLTensor *> scratch_inputs; in configure() local388 scratch_inputs.emplace_back(input_gate_out); in configure()390 scratch_inputs.emplace_back(&_cell_state_out1); in configure()391 scratch_inputs.emplace_back(forget_gate_out); in configure()392 scratch_inputs.emplace_back(output_gate_out); in configure()393 _concat_scratch_buffer.configure(compile_context, scratch_inputs, scratch_buffer, Window::DimX); in configure()