| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | hop_db.py | 16 def sample_inputs_map(opinfo, device, dtype, requires_grad, **kwargs): argument
93 def sample_inputs_cond(opinfo, device, dtype, requires_grad, **kwargs): argument
104 def sample_inputs_auto_functionalize(opinfo, device, dtype, requires_grad, **kwargs): argument
115 def sample_inputs_flex_attention(opinfo, device, dtype, requires_grad, **kwargs): argument
133 def sample_inputs_while_loop(opinfo, device, dtype, requires_grad, **kwargs): argument
|
| H A D | autograd_function_db.py | 81 def sample_inputs_numpy_cube(opinfo, device, dtype, requires_grad, **kwargs): argument
140 def sample_inputs_numpy_mul(opinfo, device, dtype, requires_grad, **kwargs): argument
145 def sample_inputs_numpy_mul_scalar(opinfo, device, dtype, requires_grad, **kwargs): argument
281 def sample_inputs_numpy_sort(opinfo, device, dtype, requires_grad, **kwargs): argument
286 def sample_inputs_numpy_take(opinfo, device, dtype, requires_grad, **kwargs): argument
417 def sample_inputs_select(opinfo, device, dtype, requires_grad, **kwargs): argument
472 def sample_inputs_forward_default_args(opinfo, device, dtype, requires_grad, **kwargs): argument
|
| H A D | common_methods_invocations.py | 3149 def error_inputs_histogramdd(opinfo, device, **kwargs): argument
3198 def error_inputs_bucketize(opinfo, device, **kwargs): argument
3408 def error_inputs_adaptive_avg_pool1d(opinfo, device, **kwargs): argument
3439 def error_inputs_adaptive_avg_pool2d(opinfo, device, **kwargs): argument
3475 def error_inputs_adaptive_avg_pool3d(opinfo, device, **kwargs): argument
3509 def error_inputs_adaptive_max_pool1d(opinfo, device, **kwargs): argument
3541 def error_inputs_adaptive_max_pool2d(opinfo, device, **kwargs): argument
3577 def error_inputs_adaptive_max_pool3d(opinfo, device, **kwargs): argument
4022 def error_inputs_conv1d(opinfo, device, **kwargs): argument
4082 def error_inputs_conv2d(opinfo, device, **kwargs): argument
[all …]
|
| H A D | custom_op_db.py | 217 def sample_inputs_numpy_nonzero(opinfo, device, dtype, requires_grad, **kwargs): argument
258 def sample_inputs_numpy_view_copy(opinfo, device, dtype, requires_grad, **kwargs): argument
301 def sample_inputs_numpy_cat(opinfo, device, dtype, requires_grad, **kwargs): argument
336 def sample_inputs_numpy_split_copy(opinfo, device, dtype, requires_grad, **kwargs): argument
431 def sample_inputs_numpy_nms(opinfo, device, dtype, requires_grad, **kwargs): argument
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/r600/sfn/ |
| H A D | sfn_instr_alugroup.cpp | 29 ASSERTED auto opinfo = alu_ops.find(instr->opcode()); in add_instruction() local
43 auto opinfo = alu_ops.find(instr->opcode()); in add_instruction() local
66 auto opinfo = alu_ops.find(instr->opcode()); in add_trans_instructions() local
|
| H A D | sfn_peephole.cpp | 109 auto opinfo = alu_ops.at(instr->opcode()); in visit() local
273 auto opinfo = alu_ops.at(new_parent->opcode()); in apply_dest_clamp() local
|
| H A D | sfn_scheduler.cpp | 1085 auto opinfo = alu_ops.find((*i)->opcode()); in collect_ready_alu_vec() local
|
| H A D | sfn_instr_alu.cpp | 2001 auto opinfo = alu_ops.at(parent->opcode()); in try_propagat_fsat64() local
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | common_utils.py | 412 def opinfo_in_dict(opinfo, d): argument
584 def check_vmap_fallback(test_case, thunk, opinfo, dry_run=False): argument
|
| H A D | test_vmap.py | 4032 def discover_variants(opinfo): argument
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/ |
| H A D | test_dtensor_ops.py | 660 def check_dtensor_func(self, test_func, opinfo, dry_run=False): argument
|