| /external/pytorch/test/functorch/ |
| D | test_minifier.py | 21 def has_mul(fx_g, inps): argument
41 def has_nans(fx_g, inps): argument
61 def inputs_returned(fx_g, inps): argument
78 def has_add(fx_g, inps): argument
105 def pass_checker(fx_g, inps): argument
|
| D | test_memory_efficient_fusion.py | 99 def run_and_compare_activation(self, fn, inps): argument
|
| D | test_aotdispatch.py | 3232 def g(inps): argument
4004 f, inps, partitioner=min_cut_rematerialization_partition, dynamic=False argument
4988 def fn(inps): argument
|
| /external/pytorch/torch/_functorch/ |
| D | fx_minifier.py | 86 def _convert_node_to_placeholder(graph, node, inps): argument
147 def dump_state(fx_g, inps): argument
176 inps, argument
213 def graph_fails(graph, inps): argument
372 def _consolidate_placeholders(cur_graph, inps): argument
|
| D | compilers.py | 55 def ts_compile(fx_g: fx.GraphModule, inps) -> Callable: argument
275 def debug_compile(fx_g, inps): argument
|
| D | eager_transforms.py | 110 def _create_differentiable(inps, level=None): argument
122 def _undo_create_differentiable(inps, level=None): argument
|
| /external/executorch/examples/models/llama/ |
| D | eval_llama_lib.py | 56 def _model_call(self, inps): argument
112 def _model_call(self, inps): argument
148 def _model_call(self, inps): argument
|
| /external/pytorch/test/dynamo/ |
| D | test_input_attr_tracking.py | 28 def grab_graph_backend(gm, inps): argument
57 def grab_graph_backend(gm, inps): argument
86 def grab_graph_backend(gm, inps): argument
|
| /external/executorch/examples/models/llama/evaluate/ |
| D | eager_eval.py | 76 def _model_call(self, inps): argument
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | peephole_alias_sensitive.cpp | 87 auto inps = node->inputs(); in runBlock() local
|
| /external/pytorch/test/higher_order_ops/ |
| D | test_with_effects.py | 53 f, inps, partitioner=min_cut_rematerialization_partition, dynamic=False argument
83 def make_inputs_non_leaves(inps): argument
|
| /external/pytorch/test/distributed/tensor/parallel/ |
| D | test_micro_pipeline_tp.py | 43 def _make_post_grad_fx(f, *inps):
|
| /external/pytorch/test/export/ |
| D | test_experimental.py | 141 def _check_equality_and_annotations(m_func, inps): argument
|
| /external/pytorch/torch/_functorch/_aot_autograd/ |
| D | utils.py | 106 def f(fx_g, inps): argument
|
| /external/pytorch/torch/csrc/jit/runtime/static/ |
| D | fusion.cpp | 47 auto inps = torch::jit::last(stack, num_inputs); in createStaticSubgraphRuntime() local
|
| /external/pytorch/benchmarks/dynamo/microbenchmarks/ |
| D | operator_inp_utils.py | 223 def deserialize_args(inps): argument
|
| /external/pytorch/test/cpp/jit/ |
| D | test_shape_analysis.cpp | 230 std::vector<at::Tensor> inps = { in TEST() local
|
| /external/pytorch/test/inductor/ |
| D | test_pad_mm.py | 434 def mm(inps, b): argument
|
| D | test_snode_runtime.py | 198 def _verify_runtime_estimation(self, fn, inps): argument
|
| D | test_cuda_repro.py | 1128 def cat(inps): argument
|
| D | test_cudagraph_trees.py | 1727 def f(inps): argument
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | gradients_test.py | 189 def model_fn(inps, init_state): argument
|
| /external/pytorch/test/ |
| D | test_expanded_weights.py | 795 def forward(self, *inps):
|
| /external/pytorch/aten/src/ATen/native/ |
| D | TensorCompare.cpp | 500 Device out_device(Args&... inps){ in out_device()
|
| /external/pytorch/torch/autograd/ |
| D | gradcheck.py | 345 def fn_pack_inps(*inps):
|