Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_dims.py38 _test_c, _parse_test, _set_pointwise_optimize = ( variable
41 _C._set_pointwise_optimize,
718 _set_pointwise_optimize(False)
721 _set_pointwise_optimize(True)
/aosp_15_r20/external/pytorch/functorch/csrc/dim/
H A Ddim.cpp3156 PyObject* _set_pointwise_optimize(PyObject * self_, in _set_pointwise_optimize() function
3217 …{"_set_pointwise_optimize", (PyCFunction)(void*) _set_pointwise_optimize, METH_FASTCALL | METH_KEY…