Home
last modified time | relevance | path

Searched refs:IsShuffle (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dop_types.h148 bool IsShuffle(const NodeDef& node);
Dop_types.cc440 bool IsShuffle(const NodeDef& node) { return node.op() == "Shuffle"; } in IsShuffle() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc1937 if (use_shape_info && (IsShuffle(*node) || IsTranspose(*node)) && in RemoveShuffleOrTranspose()