Lines Matching refs:features
32 // features: any shape.
33 // activations: same shape as "features".
34 void operator()(const Device& d, typename TTypes<T>::ConstTensor features,
45 auto too_large = features > features.constant(-threshold);
48 auto too_small = features < features.constant(threshold);
49 auto features_exp = features.exp();
51 features, // softplus(x) ~= x for x large
53 (features_exp + features.constant(T(1))).log()));
63 // features: inputs that where passed to the Softplus op.
66 typename TTypes<T>::ConstTensor features,
69 gradients / ((-features).exp() + features.constant(T(1)));
