Home
last modified time | relevance | path

Searched refs:expected_dim (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/kernels/
Dcomparisons_test.cc167 int expected_dim[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
172 input2_data, expected_data, expected_dim, output_data); in TF_LITE_MICRO_TEST()
183 int expected_dim[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
188 input2_data, expected_data, expected_dim, output_data); in TF_LITE_MICRO_TEST()
199 int expected_dim[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
203 input2_data, expected_data, expected_dim, output_data); in TF_LITE_MICRO_TEST()
214 int expected_dim[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
219 input2_data, expected_data, expected_dim, output_data); in TF_LITE_MICRO_TEST()
231 int expected_dim[] = {4, 1, 1, 2, 4}; in TF_LITE_MICRO_TEST() local
236 input2_data, expected_data, expected_dim, output_data); in TF_LITE_MICRO_TEST()
[all …]
Dpad.cc84 int expected_dim = in Prepare() local
86 TF_LITE_ENSURE_EQ(context, output_dim, expected_dim); in Prepare()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference_testutil.cc154 StringPiece expected_dim(expected_dims[j]); in InferShapes() local
167 if (expected_dim == "?") { in InferShapes()
179 } else if (absl::StartsWith(expected_dim, "d")) { in InferShapes()
181 auto v = str_util::Split(expected_dim, '|'); in InferShapes()
194 ", but should have matched one of (", expected_dim, in InferShapes()
202 if (!strings::safe_strto64(expected_dim, &value)) { in InferShapes()
204 expected_dim, "' failed to parse as int64", in InferShapes()
/external/tensorflow/tensorflow/python/keras/
Dtesting_utils.py208 for expected_dim, actual_dim in zip(expected, actual):
209 if isinstance(expected_dim, tensor_shape.Dimension):
210 expected_dim = expected_dim.value
213 if expected_dim is not None and expected_dim != actual_dim:
284 for expected_dim, actual_dim in zip(computed_output_shape,
286 if expected_dim is not None:
287 if expected_dim != actual_dim:
/external/tensorflow/tensorflow/python/framework/
Dcommon_shapes_test.py153 for expected_dim, actual_dim in zip(expected_dims, actual_dims):
154 self.assertEqual(expected_dim.value, actual_dim.value)