Searched refs:IsPolygamma (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
H A Dop_types.h91 bool IsPolygamma(const NodeDef& node);
H A Dop_types.cc224 bool IsPolygamma(const NodeDef& node) { return node.op() == "Polygamma"; } function in namespace:tensorflow::grappler
/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dlayout_optimizer.cc318 IsMinimum(node) || IsMod(node) || IsMul(node) || IsPolygamma(node) ||

Completed in 161 milliseconds