Searched refs:TestGpuSegmentedReduceKernel (Results 1 – 1 of 1) sorted by relevance
132 class TestGpuSegmentedReduceKernel : public tensorflow::OpKernel { class134 explicit TestGpuSegmentedReduceKernel( in TestGpuSegmentedReduceKernel() function in tensorflow::__anon6d95e0f40111::TestGpuSegmentedReduceKernel177 TestGpuSegmentedReduceKernel<T, Toffset, gpuprim::Sum>)