Searched defs:in_grads (Results 1 – 4 of 4) sorted by relevance
114 xla::XlaOp in_grads = ctx->Input(0); in Compile() local
928 const std::vector<Gradient*>& in_grads, absl::Span<Gradient*> out_grads) { in ForwardpropFromTape()1058 std::vector<Gradient*> in_grads; in Accumulate() local
847 def _LogOpGradients(op, out_grads, in_grads): argument
656 const Tensor& in_grads = context->input(0); in Compute() local