Searched refs:expected_dim (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/keras/_impl/keras/
H A Dtesting_utils.py123 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 Dshape_inference_testutil.cc154 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 Dhlo_sharding.cc277 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 Dcommon_shapes_test.py153 for expected_dim, actual_dim in zip(expected_dims, actual_dims):
154 self.assertEqual(expected_dim.value, actual_dim.value)

Completed in 233 milliseconds