Searched refs:expected_dim (Results 1 - 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/python/keras/_impl/keras/ |
H A D | testing_utils.py | 123 for expected_dim, actual_dim in zip(expected_output_shape, 125 if expected_dim is not None: 126 if expected_dim != actual_dim: 160 for expected_dim, actual_dim in zip(expected_output_shape, 162 if expected_dim is not None: 163 if expected_dim != actual_dim:
|
/external/tensorflow/tensorflow/core/framework/ |
H A D | shape_inference_testutil.cc | 154 StringPiece expected_dim(expected_dims[j]); 167 if (expected_dim == "?") { 179 } else if (expected_dim.starts_with("d")) { 181 auto v = str_util::Split(expected_dim, '|'); 194 ", but should have matched one of (", expected_dim, 202 if (!strings::safe_strto64(expected_dim, &value)) { 204 expected_dim, "' failed to parse as int64",
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | hlo_sharding.cc | 277 int64 expected_dim = local 279 if (tile_assignment_.dimensions()[i] != expected_dim) { 282 " expected ", expected_dim, " but got ",
|
/external/tensorflow/tensorflow/python/framework/ |
H A D | common_shapes_test.py | 153 for expected_dim, actual_dim in zip(expected_dims, actual_dims): 154 self.assertEqual(expected_dim.value, actual_dim.value)
|
Completed in 233 milliseconds