Searched refs:grad_shape_tensor (Results 1 – 1 of 1) sorted by relevance
375 const Tensor* grad_shape_tensor = nullptr; in GetAndValidateTensorInputForApplyGrad() local378 ctx->input("gradient_shape", &grad_shape_tensor)); in GetAndValidateTensorInputForApplyGrad()397 grad_idx_tensor, grad_val_tensor, grad_shape_tensor); in GetAndValidateTensorInputForApplyGrad()