Searched refs:fw_scratch_buffer (Results 1 – 1 of 1) sorted by relevance
280 std::vector<float> fw_scratch_buffer(getNumberOfElements(fw_scratch_shape_)); in Eval() local315 GetBuffer<float>(fw_output_), fw_scratch_buffer.data(), params_.time_major, in Eval()366 std::vector<_Float16> fw_scratch_buffer(getNumberOfElements(fw_scratch_shape_)); in Eval() local401 GetBuffer<_Float16>(fw_output_), fw_scratch_buffer.data(), params_.time_major, in Eval()