Searched defs:make_input (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | functorch_additional_op_db.py | 195 def make_input(shape): function
280 def make_input(shape, requires_grad=requires_grad): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/signal/ |
| H A D | reconstruction_ops_test.py | 114 def make_input(frame_length, num_frames=3): function
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/definitions/ |
| H A D | linalg.py | 1157 def make_input(): function
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_modules.py | 355 def make_input(shape, device=device, dtype=dtype, requires_grad=requires_grad): function
659 def generate_regression_criterion_inputs(make_input): argument
|
| H A D | common_methods_invocations.py | 7991 def make_input(shape): function
8125 def make_input(shape): function
8191 def make_input(shape, *, low, high): function
|
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_pytorch_onnx_onnxruntime.py | 4895 def make_input(batch_size, layers, packed_sequence): function
10175 def make_input(batch_size): function
10250 def make_input(batch_size): function
10401 def make_input(batch_size): function
|