Searched refs:IsAxis0 (Results 1 – 1 of 1) sorted by relevance
3547 if (gather_node->op() == "GatherV2" && !IsAxis0(*gather_node, 2)) in TrySimplify()3557 if (unique_node->op() == "UniqueV2" && !IsAxis0(*unique_node, 1)) in TrySimplify()3588 bool IsAxis0(const NodeDef& node, int axis_input) { in IsAxis0() function in tensorflow::grappler::__anon327bfa1e0111::SimplifyEmbeddingLookupStage