Searched refs:input_model_stream (Results 1 – 2 of 2) sorted by relevance
289 std::stringstream backport_v5_to_v4(std::stringstream& input_model_stream) { in backport_v5_to_v4() argument291 PyTorchStreamReader reader(&input_model_stream); in backport_v5_to_v4()387 std::stringstream backport_v6_to_v5(std::stringstream& input_model_stream) { in backport_v6_to_v5() argument389 std::make_shared<caffe2::serialize::IStreamAdapter>(&input_model_stream); in backport_v6_to_v5()452 std::stringstream backport_v7_to_v6(std::stringstream& input_model_stream) { in backport_v7_to_v6() argument454 std::make_shared<caffe2::serialize::IStreamAdapter>(&input_model_stream); in backport_v7_to_v6()499 std::stringstream backport_v9_to_v8(std::stringstream& input_model_stream) { in backport_v9_to_v8() argument502 torch::jit::load(input_model_stream, std::nullopt, extra_files); in backport_v9_to_v8()525 std::stringstream backport_v8_to_v7(std::stringstream& input_model_stream) { in backport_v8_to_v7() argument527 std::make_shared<caffe2::serialize::IStreamAdapter>(&input_model_stream); in backport_v8_to_v7()[all …]
597 std::stringstream& input_model_stream, in runAndCheckTorchScriptModel() argument601 auto actual_version = _get_model_bytecode_version(input_model_stream); in runAndCheckTorchScriptModel()606 Module m_mobile = load(input_model_stream); in runAndCheckTorchScriptModel()614 std::stringstream& input_model_stream, in runAndCheckBytecodeModel() argument618 auto actual_version = _get_model_bytecode_version(input_model_stream); in runAndCheckBytecodeModel()623 Module m_mobile = load(input_model_stream); in runAndCheckBytecodeModel()718 std::stringstream input_model_stream; in TEST() local720 input_model_stream, in TEST()743 input_model_stream, in TEST()