Home
last modified time | relevance | path

Searched defs:RowMaxReduction (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_xent_op_gpu.cu.cc36 struct RowMaxReduction<GPUDevice, T> { struct
41 static inline void Compute(OpKernelContext* ctx, in Compute()
Dsparse_xent_op.h133 struct RowMaxReduction { struct
138 static inline void Compute(OpKernelContext* ctx, in Compute()