Searched refs:IsShuffle (Results 1 – 3 of 3) sorted by relevance
167 bool IsShuffle(const NodeDef& node);
519 bool IsShuffle(const NodeDef& node) { return node.op() == "Shuffle"; } in IsShuffle() function
2136 if (!use_shape_info || !(IsShuffle(*node) || IsTranspose(*node))) in RemoveShuffleOrTranspose()