Home
last modified time | relevance | path

Searched refs:pytorch_version (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/scripts/release/promote/
H A Dconda_to_conda.sh16 pytorch_version="$(get_pytorch_version)"
23 …conda search -q "${PYTORCH_CONDA_FROM}::${PACKAGE_NAME}==${pytorch_version}" -c "${PYTORCH_CONDA_F…
26 …/${PYTORCH_CONDA_FROM}/${PACKAGE_NAME}/${pytorch_version}/download/${platform}/${PACKAGE_NAME}-${p…
H A Dcommon_utils.sh40 pytorch_version=$(get_pytorch_version)
55 --include "*${package_name}-${pytorch_version}*" \
H A Dwheel_to_pypi.sh12 pytorch_version="$(get_pytorch_version)"
24 | grep "${PACKAGE_NAME}-${pytorch_version}${VERSION_SUFFIX}-" \
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/optim/
H A Dserialize.h167 IValue pytorch_version; in serialize() local
168 archive.read("pytorch_version", pytorch_version); in serialize()
169 TORCH_INTERNAL_ASSERT(pytorch_version.toStringRef() == "1.5.0"); in serialize()
/aosp_15_r20/external/pytorch/torch/csrc/api/src/optim/
H A Dsgd.cpp114 IValue pytorch_version; in load() local
115 if (archive.try_read("pytorch_version", pytorch_version)) { in load()
H A Dadagrad.cpp131 IValue pytorch_version; in load() local
132 if (archive.try_read("pytorch_version", pytorch_version)) { in load()
H A Drmsprop.cpp148 IValue pytorch_version; in load() local
149 if (archive.try_read("pytorch_version", pytorch_version)) { in load()
H A Dadam.cpp150 IValue pytorch_version; in load() local
151 if (archive.try_read("pytorch_version", pytorch_version)) { in load()
H A Dadamw.cpp151 IValue pytorch_version; in load() local
152 if (archive.try_read("pytorch_version", pytorch_version)) { in load()
H A Dlbfgs.cpp622 IValue pytorch_version; in load() local
623 if (archive.try_read("pytorch_version", pytorch_version)) { in load()
/aosp_15_r20/external/pytorch/benchmarks/distributed/ddp/
H A DREADME.md121 pytorch_version: 1.4.0a0+05140f0 vs 1.4.0a0+05140f0