Home
last modified time | relevance | path

Searched refs:input_model_stream (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/compatibility/
H A Dbackport_manager.cpp289 std::stringstream backport_v5_to_v4(std::stringstream& input_model_stream) { in backport_v5_to_v4() argument
291 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() argument
389 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() argument
454 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() argument
502 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() argument
527 std::make_shared<caffe2::serialize::IStreamAdapter>(&input_model_stream); in backport_v8_to_v7()
[all …]
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_lite_interpreter.cpp597 std::stringstream& input_model_stream, in runAndCheckTorchScriptModel() argument
601 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() argument
618 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() local
720 input_model_stream, in TEST()
743 input_model_stream, in TEST()