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

Function MoveValues

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

Source from the content-addressed store, hash-verified

99}
100template <typename TIndex>
101void MoveValues(const GPUDevice& d, const TIndex* keys, const TIndex* values,
102 const int64 size, TIndex* out) {
103 GpuLaunchConfig config = GetGpuLaunchConfig(size, d);
104 MoveValuesKernel<<<config.block_count, config.thread_per_block, 0,
105 d.stream()>>>(keys, values, size, out);
106}
107
108template <typename TIndex>
109void MoveValues(const GPUDevice& d, const TIndex* keys, const TIndex* values,

Callers 1

ComputeAsyncMethod · 0.70

Calls 2

GetGpuLaunchConfigFunction · 0.85
streamMethod · 0.45

Tested by

no test coverage detected