5c42053fcd6821eb39eded56526cfe5f2070418a,sandbox/rocky/tf/optimizers/conjugate_gradient_optimizer.py,ConjugateGradientOptimizer,update_opt,#ConjugateGradientOptimizer#Any#Any#Any#Any#Any#Any#,166

Before Change

        

        inputs = tuple(inputs)
        if extra_inputs is None:
            extra_inputs = tuple()
        else:
            extra_inputs = tuple(extra_inputs)

        constraint_term, constraint_value = leq_constraint

        params = target.get_params(trainable=True)
        grads = tf.gradients(loss, xs=params)

After Change

            self.opt_fun = ext.lazydict(
                f_Hx_plain=lambda: tensor_utils.compile_function(
                    inputs=inputs + xs,
                    outputs=Hx_plain(),
                    log_name="f_Hx_plain",
                ),
            )

Instances