| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | compile_fx.py | 648 def compiled_artifact(new_inputs): argument
950 def run(new_inputs): argument
996 def run(new_inputs): argument
1101 def run(new_inputs): argument
1124 def run(new_inputs): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | einsum_spmd_expander.cc | 55 std::vector<mlir::Value> new_inputs; in ExpandOp() local
390 Layout& einsum_layout, std::vector<mlir::Value>& new_inputs) { in MaybeRelayoutInputs()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | concat_opt.cpp | 119 std::vector<Value*> new_inputs = { in handleCat() local
164 std::vector<Value*> new_inputs = { in handleCat() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/fallback/ |
| H A D | op_kernel_runner_test.cc | 66 gtl::InlinedVector<TensorValue, 4> new_inputs{TensorValue(&c)}; in TEST() local
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/backends/ |
| H A D | cudagraphs.py | 245 def run(*new_inputs):
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | reduce_decomposer.cc | 38 std::vector<HloInstruction*> new_inputs; in HandleReduce() local
|
| H A D | hlo_dataflow_analysis.cc | 494 std::vector<HloValue*> new_inputs; in Phi() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | symbolic_script.cpp | 1585 std::vector<Value*> new_inputs = forward_tuple->inputs().vec(); in loadModule() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | control_flow_deps_to_chains.cc | 143 std::vector<Node*> new_inputs; in Run() local
|
| /aosp_15_r20/external/mesa3d/src/compiler/nir/ |
| H A D | nir_lower_io_to_temporaries.c | 43 struct exec_list new_inputs; member
|
| H A D | nir_lower_io_to_vector.c | 423 nir_variable *new_inputs[MAX_SLOTS][4] = { { 0 } }; in nir_lower_io_to_vector_impl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | region_control_flow_to_functional.cc | 404 auto new_inputs = llvm::to_vector<4>(while_region.getOperands()); in ConvertWhileOp() local
|
| H A D | tpu_variable_runtime_reformatting.cc | 225 MutableArrayRef<TF::VarHandleOp> new_inputs, in AddInputsToReplicateOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/ |
| H A D | pfor.py | 614 new_inputs, not_all_done): argument
4846 new_inputs, not_all_done): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | engine.cpp | 1005 variable_list new_inputs = inputs.buffer; in evaluate_function() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/python/ |
| H A D | python_ir.cpp | 390 [](Graph& g, const std::vector<size_t>& new_inputs) { in initPythonIRBindings()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/ir/ |
| H A D | ir.cpp | 1572 std::vector<Value*> new_inputs; in permuteInputs() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | loopnest.cpp | 550 std::vector<ExprPtr> new_inputs; in vectorize_inputs() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/ |
| H A D | ir_emitter.cpp | 4063 auto new_inputs = List<Expr>::create(apply.range(), {ll}); in emitApplySpecialFormForDict() local
|