Searched defs:approximate (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ |
| D | test_ops_gelu.py | 32 def construct(self, x, approximate): argument
43 def test_ops_forward(context_mode, approximate): argument
135 def test_ops_backward(context_mode, approximate): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/fusion/ |
| D | activation.cc | 66 void Activation::set_approximate(bool approximate) { (void)this->AddAttr(kApproximate, api::MakeVal… in set_approximate()
74 const ActivationType &activation_type, bool approximate) { in Init()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/common/ops/operator_populate/ |
| D | activation_populate.cc | 87 mindspore::ValuePtr approximate = base_operator->GetPrim()->GetAttr(kApproximate); in PopulateActivationOpParameter() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | activation_fp32.c | 200 int Gelu(const float *src, int length, float *dst, bool approximate) { in Gelu()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/ |
| D | activation_fp16.c | 240 int GeluFp16(const float16_t *src, int length, float16_t *dst, bool approximate) { in GeluFp16()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/layer/ |
| D | activation.py | 925 def __init__(self, approximate=True): argument
|
| /third_party/mesa3d/include/tensorflow/lite/core/c/ |
| D | builtin_op_data.h | 530 bool approximate; member
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/ |
| D | nn_func.py | 7027 def gelu(input, approximate='none'): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/mindir/src/ |
| D | mindir.cc | 26 float max_val, bool approximate) { in MindIR_Activation_CreatePrimitive()
197 void MindIR_Activation_SetApproximate(PrimitivePtr *primitive, bool approximate) { in MindIR_Activation_SetApproximate()
|