Searched defs:sample_inputs (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | impl.cpp | 159 std::vector<IValue> sample_inputs) { in OptimizeGraph()
271 std::vector<IValue> sample_inputs) { in PrepareGraphForStaticModule()
293 std::vector<IValue> sample_inputs) { in PrepareForStaticModule()
322 std::vector<IValue> sample_inputs) { in PrepareForStaticModule()
531 std::vector<IValue> sample_inputs) in StaticModule()
540 std::vector<IValue> sample_inputs) in StaticModule()
|
| H A D | fusion.cpp | 337 std::vector<IValue> sample_inputs) { in performTensorExprFusion()
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/definitions/ |
| H A D | sparse.py | 53 def _sample_inputs_sparse( argument
89 def _apply_requires_grad_to_samples(sample_inputs): argument
|
| /aosp_15_r20/packages/modules/Virtualization/libs/dice/sample_inputs/src/ |
| D | lib.rs | 24 mod sample_inputs; module
|
| /aosp_15_r20/external/executorch/backends/apple/mps/test/ |
| H A D | test_mps_utils.py | 100 def dump_bundled_program(sample_inputs, expected_output, executorch_program, func_name): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_masked.py | 300 def test_mask_layout(self, layout, device, dtype, op, sample_inputs): argument
|
| H A D | test_expanded_weights.py | 1081 def supported_inputs(op, sample_inputs, supported_inputs=True): argument
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_aot_autograd.py | 946 def mini_backend(gm, sample_inputs): argument
|
| H A D | test_export.py | 1173 def compiler(gm, sample_inputs): argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/ |
| H A D | core.py | 1228 def sample_inputs(self, device, dtype, requires_grad=False, **kwargs): member in OpInfo
|