Searched defs:graph_fn (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | sharding_util_ops_test.py | 66 def testSplitDimensionZero(self, graph_fn): argument
80 def testSplitDimensionNegative(self, graph_fn): argument
94 def testNumOutputsMismatch(self, graph_fn): argument
108 def testPaddingsLengthMismatch(self, graph_fn): argument
123 def testPaddingsNegative(self, graph_fn): argument
138 def testInputRankSplitMismatch(self, graph_fn): argument
153 def testDimNotEvenlySplit(self, graph_fn): argument
167 def testDimWithPaddingNotEvenlySplit(self, graph_fn): argument
182 def testNoSplits(self, graph_fn): argument
197 def testNoSplitsWithPadding(self, graph_fn): argument
[all …]
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | batch_matmul_test.py | 34 def BuildParams(self, graph_fn, dtype, input_shapes, output_shapes): argument
|
| D | tf_trt_integration_test_base.py | 177 def BuildParams(self, graph_fn, dtype, input_shapes, output_shapes): argument
208 def BuildParamsWithMask(self, graph_fn, dtype, input_shapes, output_shapes, argument
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | meta_graph_test.py | 1003 def _testExportImportAcrossScopes(self, graph_fn, use_resource): argument
|
| /external/tensorflow/tensorflow/python/training/ |
| D | saver_test.py | 2275 def _testGradientSerDes(self, graph_fn): argument
|
| /external/tensorflow/tensorflow/python/autograph/impl/ |
| D | api_test.py | 831 def graph_fn(): function
|