Searched defs:expected_output_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | convolutional_test.py | 40 def _run_test(self, kwargs, expected_output_shape): argument
52 def _run_test_extra_batch_dim(self, kwargs, expected_output_shape): argument
98 def test_conv1d(self, kwargs, expected_output_shape, requires_gpu=False): argument
177 def _run_test(self, kwargs, expected_output_shape, spatial_shape=(7, 6)): argument
197 expected_output_shape, argument
255 expected_output_shape=None, argument
309 def _run_test(self, kwargs, expected_output_shape, validate_training=True): argument
326 expected_output_shape, argument
370 def test_conv3d(self, kwargs, expected_output_shape=None, requires_gpu=False): argument
496 def _run_test(self, kwargs, expected_output_shape): argument
[all …]
|
| D | einsum_dense_test.py | 229 expected_output_shape): argument
240 expected_output_shape): argument
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | testing_utils.py | 99 expected_output_shape=None, argument
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | functional_test.py | 1517 def _testShapeInference(self, model, input_shape, expected_output_shape): argument
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 5270 std::vector<int> expected_output_shape; in TEST_P() member
|