Searched refs:cpp_inputs (Results 1 – 2 of 2) sorted by relevance
45 std::map<exec_aten::string_view, exec_aten::Tensor> cpp_inputs; in PySGD() local58 cpp_inputs.insert({v, *params_.back()}); in PySGD()60 cpp_inputs.insert({v, pair.second}); in PySGD()64 cpp_inputs, in PySGD()78 std::map<exec_aten::string_view, exec_aten::Tensor> cpp_inputs; in step() local99 cpp_inputs.insert({pair.first.c_str(), *et_tensors.back()}); in step()101 cpp_inputs.insert({pair.first.c_str(), at_tensor}); in step()105 auto err = sgd_->step(cpp_inputs); in step()
673 std::vector<EValue> cpp_inputs; in run_method() local674 cpp_inputs.reserve(inputs_size); in run_method()740 cpp_inputs.push_back(evalue); in run_method()742 cpp_inputs.push_back(EValue()); in run_method()744 cpp_inputs.push_back(EValue(py::cast<bool>(python_input))); in run_method()746 cpp_inputs.push_back(EValue(py::cast<int64_t>(python_input))); in run_method()761 module_->run_method(method_name, cpp_inputs, output_storage_spans); in run_method()