Searched refs:generate_vmap_inputs (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_ops.py | 17 generate_vmap_inputs,
1716 generator = generate_vmap_inputs(
2651 generator = generate_vmap_inputs(args, kwargs, batch_size=B)
2678 generator = generate_vmap_inputs(batched_args, {})
2704 generator = generate_vmap_inputs(args, kwargs, batch_size=B)
2708 generator = generate_vmap_inputs(batched_args, kwargs)
2747 generator = generate_vmap_inputs(args, kwargs, batch_size=B)
2784 generator = generate_vmap_inputs(args, kwargs, batch_size=B)
2818 generator = generate_vmap_inputs(args, kwargs, batch_size=B)
2822 generator = generate_vmap_inputs(batched_args, kwargs)
[all …]
|
| H A D | common_utils.py | 244 def generate_vmap_inputs( function
394 generator = generate_vmap_inputs(
|
| H A D | test_vmap.py | 26 generate_vmap_inputs,
4123 for batched_args, in_dims, _ in generate_vmap_inputs(args, {}):
4164 for batched_args, in_dims, _ in generate_vmap_inputs(
5080 ) in generate_vmap_inputs(args, {}):
|