Home | History | Annotate | Download | only in kernels

Lines Matching defs:activations

667       auto activations = out_image.shaped<T, 2>({nodes * batch, depth});
678       auto shard = [this, activations, in_shaped, grads_shaped, out_shaped,
693                       activations(i, j) / norm;
1236     auto activations = orig_output_tensor.shaped<T, 2>({nodes * batch, depth});
1248     auto shard = [this, activations, in_shaped, grads_shaped, out_shaped,
1263                     activations(i, j) / norm;