Searched refs:GetFloatAttr (Results 1 – 2 of 2) sorted by relevance
49 float GetFloatAttr(const string& name) { in GetFloatAttr() function in tensorflow::LRNFloatTest64 const float bias = GetFloatAttr("bias"); in Compare()65 const float alpha = GetFloatAttr("alpha"); in Compare()66 const float beta = GetFloatAttr("beta"); in Compare()
98 float GetFloatAttr(const NodeDef& node, const std::string& attr_name) { in GetFloatAttr() function1119 minmax.min = GetFloatAttr(node, "min"); in ConvertFakeQuantWithMinMaxArgs()1120 minmax.max = GetFloatAttr(node, "max"); in ConvertFakeQuantWithMinMaxArgs()1238 softmax->beta = GetFloatAttr(node, "_softmax_beta"); in ConvertSoftmaxOperator()1256 lrn->bias = GetFloatAttr(node, "bias"); in ConvertLRNOperator()1257 lrn->alpha = GetFloatAttr(node, "alpha"); in ConvertLRNOperator()1258 lrn->beta = GetFloatAttr(node, "beta"); in ConvertLRNOperator()1853 {GetFloatAttr(node, "epsilon")}); in ConvertFusedBatchNormOperator()2518 op->alpha = GetFloatAttr(node, "alpha"); in ConvertLeakyReluOperator()