Home
last modified time | relevance | path

Searched defs:input_vars (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_function.cpp606 const variable_list& input_vars, in _wrap_outputs()
857 variable_list input_vars; member
977 const variable_list& input_vars) { in _trace_pre_record()
1008 const variable_list& input_vars, in _trace_post_record()
H A Dcustom_function.h330 variable_list input_vars; variable
H A Dcustom_function.cpp439 const variable_list& input_vars, in _wrap_outputs()
H A Dvariable.cpp591 std::vector<torch::autograd::Variable> input_vars( in _backward() local
/aosp_15_r20/external/pytorch/test/cpp/lazy/
H A Dtest_lazy_ops_util.cpp135 std::vector<torch::Tensor> input_vars; in TestBackward() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dgradients_test.py1563 def TestFn(inputs, input_vars): argument
1566 def TestFnSeq(inputs, input_vars): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
H A Dconv_emitter.cc312 std::vector<mlir::Value> input_vars; in CreateNaiveMlirConv() local
/aosp_15_r20/external/mesa3d/src/mesa/program/
H A Dprog_to_nir.c56 nir_variable *input_vars[VARYING_SLOT_MAX]; member
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dregister_prim_ops_fulljit.cpp236 std::vector<torch::autograd::Variable> input_vars( in __anon1bf66a481902() local
/aosp_15_r20/external/mesa3d/src/compiler/nir/
H A Dnir_linking_helpers.c963 nir_variable *input_vars[MAX_VARYING][4] = { 0 }; in nir_link_xfb_varyings() local