Searched refs:IsTranspose (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer.cc | 167 if (!GetValueAttrFromConstInputNode(fanout_transpose, IsTranspose, 1, in IsCancellableConstPermTransposeNodePair()
172 if (!GetValueAttrFromConstInputNode(fanin_transpose, IsTranspose, 1, in IsCancellableConstPermTransposeNodePair()
284 if (!IsTranspose(*transpose_after->node())) continue; in EraseCancellableNodesAroundPad()
297 if (!IsTranspose(*transpose_before->node())) continue; in EraseCancellableNodesAroundPad()
318 if (!GetValueAttrFromConstInputNode(*transpose_after, IsTranspose, 1, in EraseCancellableNodesAroundPad()
334 if (!GetValueAttrFromConstInputNode(*extra_transpose, IsTranspose, 1, in EraseCancellableNodesAroundPad()
|
| D | arithmetic_optimizer.cc | 1120 return IsTranspose(*node) || IsConjugateTranspose(*node); in IsSupported()
2392 return IsConj(*node) || IsTranspose(*node) || IsConjugateTranspose(*node); in IsSupported()
2406 if ((IsTranspose(*transpose_op) || IsConjugateTranspose(*transpose_op)) && in TrySimplify()
|
| D | generic_layout_optimizer_transposer.cc | 1052 if (!GetValueAttrFromConstInputNode(node, IsTranspose, 1, &tensor)) { in IsValidConstPermTransposeNode()
|
| D | constant_folding.cc | 2136 if (!use_shape_info || !(IsShuffle(*node) || IsTranspose(*node))) in RemoveShuffleOrTranspose()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | op_types.h | 198 bool IsTranspose(const NodeDef& node);
|
| D | op_types.cc | 630 bool IsTranspose(const NodeDef& node) { return node.op() == "Transpose"; } in IsTranspose() function
|