Home
last modified time | relevance | path

Searched refs:MyStatefulReluGPU (Results 1 – 2 of 2) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/example/extensions/lib_custom_op/
H A Drelu_lib.cu150 class MyStatefulReluGPU : public CustomStatefulOp { class
152 explicit MyStatefulReluGPU(const std::unordered_map<std::string, std::string>& attrs) in MyStatefulReluGPU() function in MyStatefulReluGPU
164 ~MyStatefulReluGPU() {} in ~MyStatefulReluGPU()
183 *op_inst = new MyStatefulReluGPU(attrs); in createOpStateGPU()
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/example/extensions/lib_custom_op/
H A Drelu_lib.cu150 class MyStatefulReluGPU : public CustomStatefulOp { class
152 explicit MyStatefulReluGPU(const std::unordered_map<std::string, std::string>& attrs) in MyStatefulReluGPU() function in MyStatefulReluGPU
164 ~MyStatefulReluGPU() {} in ~MyStatefulReluGPU()
183 *op_inst = new MyStatefulReluGPU(attrs); in createOpStateGPU()