(activation)
| 168 | projection = tf_layers.Dense(1) |
| 169 | |
| 170 | def model_fn(activation): |
| 171 | for layer in layers: |
| 172 | activation = layer(activation) |
| 173 | activation = projection(activation) |
| 174 | activation = nn.l2_loss(activation) |
| 175 | return gradient_ops.gradients(activation, variables.trainable_variables()) |
| 176 | |
| 177 | def loop_fn(i): |
| 178 | return model_fn(array_ops.expand_dims(array_ops.gather(inp, i), 0)) |
no test coverage detected