Searched refs:sample_inputs_numpy_cube (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | autograd_function_db.py | 81 def sample_inputs_numpy_cube(opinfo, device, dtype, requires_grad, **kwargs): function
511 sample_inputs_func=sample_inputs_numpy_cube,
521 sample_inputs_func=sample_inputs_numpy_cube,
539 sample_inputs_func=sample_inputs_numpy_cube,
576 sample_inputs_func=sample_inputs_numpy_cube,
613 sample_inputs_func=sample_inputs_numpy_cube,
|
| H A D | custom_op_db.py | 13 sample_inputs_numpy_cube,
448 sample_inputs_func=sample_inputs_numpy_cube,
|