MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / _is_transpose

Function _is_transpose

tensorflow/python/grappler/layout_optimizer_test.py:195–199  ·  view source on GitHub ↗
(node)

Source from the content-addressed store, hash-verified

193
194
195def _is_transpose(node):
196 return node.endswith('TransposeNHWCToNCHW-LayoutOptimizer') or node.endswith(
197 'TransposeNCHWToNHWC-LayoutOptimizer') or node.endswith(
198 'TransposeNDHWCToNCDHW-LayoutOptimizer') or node.endswith(
199 'TransposeNCDHWToNDHWC-LayoutOptimizer')
200
201
202def _is_permute(node):

Callers 15

testTwoConvLayersMethod · 0.85
testSplitWithNonConstAxisMethod · 0.85
testSplitVWithNonConstAxisMethod · 0.85
testPadWithConstPaddingsMethod · 0.85
testReduceSumMethod · 0.85
testCastMethod · 0.85
testSqueezeMethod · 0.85
testSqueezeAlongHWMethod · 0.85
testSqueezeAlongNHWMethod · 0.85
testReduceSumAlongHWCMethod · 0.85
testReduceSumAlongNHWMethod · 0.85
testReduceSumAlongCMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected