Home
last modified time | relevance | path

Searched refs:kReluGradInputTensorNum (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dderelu_fusion.cc34 CheckCNodeInputSize(relu_grad, kReluGradInputTensorNum); in GetRelu()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Drelu_v2_pass.cc36 CheckCNodeInputSize(relu_grad, kReluGradInputTensorNum); in GetRelu()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.h36 constexpr size_t kReluGradInputTensorNum = 2; variable