Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A 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
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_engine.cpp288 const Variable& grad_var = THPVariable_Unpack(grad); in THPEngine_run_backward() local
H A DFunctionsManual.cpp1868 auto grad_var = (grad / (result * 2)).masked_fill_(result == 0, 0); in std_backward() local