Home
last modified time | relevance | path

Searched refs:group_vals (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_softmax_op.cc102 const auto group_vals = g.values<T>(); in Compute() local
103 const int group_size = group_vals.size(); in Compute()
106 tmp_scalar.device(context->eigen_device<Device>()) = group_vals.maximum(); in Compute()
110 tmp.device(device) = (group_vals - tmp.constant(group_max)).exp(); in Compute()