MCPcopy Create free account
hub / github.com/dromara/Omega-AI / max_gpu

Method max_gpu

src/main/java/com/omega/engine/ad/op/gpu/OPKernel.java:1432–1461  ·  view source on GitHub ↗
(Tensor a, Tensor y, int axis)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

maxMethod · 0.80

Calls 4

checkCUDAMethod · 0.95
CAFFE_GET_BLOCKSMethod · 0.95
getGpuDataMethod · 0.80
getDataLengthMethod · 0.80

Tested by

no test coverage detected