Home
last modified time | relevance | path

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

/external/pytorch/test/cpp/lazy/
Dtest_lazy_ops_util.cpp135 std::vector<torch::Tensor> input_vars; in TestBackward() local
/external/pytorch/torch/csrc/autograd/
Dcustom_function.h330 variable_list input_vars; variable
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()
Dcustom_function.cpp439 const variable_list& input_vars, in _wrap_outputs()
Dvariable.cpp591 std::vector<torch::autograd::Variable> input_vars( in _backward() local
/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
Dconv_emitter.cc312 std::vector<mlir::Value> input_vars; in CreateNaiveMlirConv() local
/external/tensorflow/tensorflow/python/ops/
Dgradients_test.py1563 def TestFn(inputs, input_vars): argument
1566 def TestFnSeq(inputs, input_vars): argument
/external/pytorch/torch/csrc/jit/runtime/
Dregister_prim_ops_fulljit.cpp236 std::vector<torch::autograd::Variable> input_vars( in __anon992a14c81902() local
/external/mesa3d/src/compiler/nir/
Dnir_linking_helpers.c962 nir_variable *input_vars[MAX_VARYING][4] = { 0 }; in nir_link_xfb_varyings() local