MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / CompareAdjacent

Function CompareAdjacent

tensorflow/core/kernels/unique_ali_op_gpu.cu.cc:123–129  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

121}
122template <typename T, typename TIndex>
123void CompareAdjacent(const GPUDevice& d, const T* in, const int64 size,
124 TIndex* out) {
125 GpuLaunchConfig config = GetGpuLaunchConfig(size, d);
126 CompareAdjacentKernel<
127 T><<<config.block_count, config.thread_per_block, 0, d.stream()>>>(
128 in, size, out);
129}
130template <typename T, typename TIndex>
131class UniqueAliV2GpuOp : public AsyncOpKernel {
132 public:

Callers

nothing calls this directly

Calls 2

GetGpuLaunchConfigFunction · 0.85
streamMethod · 0.45

Tested by

no test coverage detected