Searched defs:orig_inputs (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/ |
| H A D | runtime_wrappers.py | 155 def __call__(self, orig_inputs, fw_outs, out): argument
176 def __call__(self, orig_inputs, fw_outs, out): argument
192 def __call__(self, orig_inputs, fw_outs, out): argument
213 def __call__(self, orig_inputs, fw_outs, out): argument
|
| /aosp_15_r20/external/pytorch/torch/fx/passes/ |
| H A D | split_utils.py | 50 orig_inputs: List = field(default_factory=list) variable in Component
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/ |
| H A D | utils.cc | 225 std::vector<int> orig_inputs; in IsNodeSupported() local
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | autograd_function.py | 168 outputs, unwrapped_inputs, orig_inputs, wrap_fn, out_dims=NO_OUT_DIMS argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | distributed_tpu_rewrite_pass.cc | 1235 const std::vector<NodeOut>& orig_inputs, Graph* graph, in CreateConcatNodesForRetval()
1286 const std::vector<NodeOut>& orig_inputs, absl::string_view device, in CreateXlaConcatNode()
3760 std::vector<NodeOut> orig_inputs; in BuildExecuteNodes() local
3847 std::vector<NodeOut> orig_inputs; in BuildExecuteNodes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/ |
| H A D | training_v1.py | 2542 def _compile_from_inputs(self, all_inputs, target, orig_inputs, orig_target): argument
|