Home
last modified time | relevance | path

Searched defs:grad_var (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtensor_array_ops_decomposition.cc402 Value grad_var; in HandleTensorArrayGradV3Op() local
614 Value grad_var; in HandleWhileOp() local
681 Value grad_var; in HandleIfOp() local
741 Value grad_var; in HandlePartitionedCallOp() local
/external/pytorch/torch/csrc/autograd/
Dpython_engine.cpp288 const Variable& grad_var = THPVariable_Unpack(grad); in THPEngine_run_backward() local
DFunctionsManual.cpp1868 auto grad_var = (grad / (result * 2)).masked_fill_(result == 0, 0); in std_backward() local