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

Function RangeInit

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

Source from the content-addressed store, hash-verified

91}
92template <typename TIndex>
93void RangeInit(const GPUDevice& d, const TIndex start, const TIndex delta,
94 const int64 size, TIndex* out) {
95 GpuLaunchConfig config = GetGpuLaunchConfig(size, d);
96 RangeInitKernel<TIndex>
97 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>(
98 start, delta, size, out);
99}
100template <typename TIndex>
101void MoveValues(const GPUDevice& d, const TIndex* keys, const TIndex* values,
102 const int64 size, TIndex* out) {

Callers 1

ComputeAsyncMethod · 0.70

Calls 2

GetGpuLaunchConfigFunction · 0.85
streamMethod · 0.45

Tested by

no test coverage detected