Searched defs:unpack_fn (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | unpack_quantized_weights.cpp | 261 const std::string& unpack_fn, in unpackQuantizedWeightsHelper()
|
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_quantized_op.py | 4507 …def _test_embedding_bag_unpack_impl(self, pack_fn, unpack_fn, bit_rate, optimized_qparams, weights… argument
4548 …def _test_embedding_bag_unpack_fn(self, pack_fn, unpack_fn, num_embeddings, embedding_dim, bit_rat… argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | python_function.cpp | 77 const std::function<PyObject*(const Variable&)>& unpack_fn) { in unpack_saved_variables()
|