Searched refs:GetFloatAttr (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lrn_op_test.cc | 46 float GetFloatAttr(const string& name) { in GetFloatAttr() function in tensorflow::LRNFloatTest
61 const float bias = GetFloatAttr("bias"); in Compare()
62 const float alpha = GetFloatAttr("alpha"); in Compare()
63 const float beta = GetFloatAttr("beta"); in Compare()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow.cc | 96 float GetFloatAttr(const NodeDef& node, const string& attr_name) { in GetFloatAttr() function
868 minmax.min = GetFloatAttr(node, "min"); in ConvertFakeQuantWithMinMaxArgs()
869 minmax.max = GetFloatAttr(node, "max"); in ConvertFakeQuantWithMinMaxArgs()
1001 lrn->bias = GetFloatAttr(node, "bias"); in ConvertLRNOperator()
1002 lrn->alpha = GetFloatAttr(node, "alpha"); in ConvertLRNOperator()
1003 lrn->beta = GetFloatAttr(node, "beta"); in ConvertLRNOperator()
1730 model, node.name() + "_epsilon_array", {GetFloatAttr(node, "epsilon")}); in ConvertFusedBatchNormOperator()
2368 op->alpha = GetFloatAttr(node, "alpha"); in ConvertLeakyReluOperator()
|