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