Home | History | Annotate | Download | only in kernels

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,
37 features / (features.abs() + features.constant(T(1)));
47 // features: inputs that were passed to the Softsign op.
50 typename TTypes<T>::ConstTensor features,
53 gradients / (features.abs() + features.constant(T(1))).square();