Home
last modified time | relevance | path

Searched refs:raw_outputs (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/torch/csrc/autograd/
Dcustom_function.cpp31 const at::ArrayRef<std::optional<Variable>> raw_outputs, in _process_forward_mode_AD() argument
112 if (!raw_outputs[i].has_value()) { in _process_forward_mode_AD()
117 auto out_tensor_impl = raw_outputs[i].value().unsafeGetTensorImpl(); in _process_forward_mode_AD()
120 isDifferentiableType(raw_outputs[i].value().scalar_type())); in _process_forward_mode_AD()
260 const at::ArrayRef<std::optional<Variable>> raw_outputs, in _process_backward_mode_ad() argument
264 auto num_outputs = raw_outputs.size(); in _process_backward_mode_ad()
356 if (!raw_outputs[i].has_value()) { in _process_backward_mode_ad()
366 Variable var = raw_outputs[i].value(); in _process_backward_mode_ad()
442 const at::ArrayRef<std::optional<Variable>> raw_outputs, in _wrap_outputs() argument
457 raw_outputs, in _wrap_outputs()
[all …]
Dcustom_function.h23 const at::ArrayRef<std::optional<Variable>> raw_outputs,
/external/pytorch/torch/autograd/
Dgradcheck.py552 raw_outputs = _as_tuple(fn(*dual_inputs))
553 dual_outputs = filter(_is_float_or_complex_tensor, raw_outputs)
577 raw_outputs = _as_tuple(fn(*dual_inputs))
578 dual_outputs = filter(_is_float_or_complex_tensor, raw_outputs)
1262 raw_outputs = _as_tuple(func(*dual_inputs))
1263 dual_outputs1 = filter(_is_float_or_complex_tensor, raw_outputs)
1267 raw_outputs = _as_tuple(func(*dual_inputs))
1268 dual_outputs2 = filter(_is_float_or_complex_tensor, raw_outputs)
/external/skia/gn/
Dgn_to_cmake.py401 raw_outputs = target.properties.get('outputs', [])
405 for output in raw_outputs: