Searched defs:grad_tensor (Results 1 – 4 of 4) sorted by relevance
122 const Tensor* grad_tensor; in GetAndValidateTensorInputForApplyGrad() local
691 const Tensor& grad_tensor = in Compute() local815 const Tensor& grad_tensor, in SanityCheckInputs()
578 const Tensor& grad_tensor = in Compute() local
366 const tensorflow::Tensor& grad_tensor, in create_grad_tensor()413 const tensorflow::Tensor& grad_tensor = context->input(0); in Compute() local