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

Method assign

tensorflow/core/kernels/topk_op_gpu.h:137–137  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

135 }
136
137 __device__ void assign(int i, const Entry& entry) { data[i] = entry; }
138
139 __device__ void push_up(int i) {
140 int child = i;

Callers 15

ConvertTfliteConstTensorFunction · 0.45
ParseSwitchOpFunction · 0.45
ParseMergeOpFunction · 0.45
ParseEnterOpFunction · 0.45
ParseLoopCondOpFunction · 0.45
buildMethod · 0.45
runOnFunctionMethod · 0.45
AddBackedgeMethod · 0.45
LhloFuseLinalgMethod · 0.45
matchAndRewriteMethod · 0.45
gMethod · 0.45

Calls

no outgoing calls

Tested by 15

gMethod · 0.36
testWriteEmptyShapeMethod · 0.36
testReadWriteMethod · 0.36
__setitem__Method · 0.36
testUninitializedMethod · 0.36
tfadd_with_ckptFunction · 0.36
tfadd_with_ckpt_saverFunction · 0.36
TESTFunction · 0.36
TESTFunction · 0.36
InitializeMethod · 0.36
TEST_FFunction · 0.36