| /external/tensorflow/tensorflow/compiler/jit/ |
| D | test_util.cc | 25 std::map<string, std::vector<PartialTensorShape>> expected_shapes) { in ShapeAnnotationsMatch() argument
33 auto it = expected_shapes.find(node->name()); in ShapeAnnotationsMatch()
34 if (it != expected_shapes.end()) { in ShapeAnnotationsMatch()
42 expected_shapes.erase(it); in ShapeAnnotationsMatch()
45 if (!expected_shapes.empty()) { in ShapeAnnotationsMatch()
47 missing.reserve(expected_shapes.size()); in ShapeAnnotationsMatch()
48 for (const auto& entry : expected_shapes) { in ShapeAnnotationsMatch()
|
| D | test_util.h | 39 std::map<string, std::vector<PartialTensorShape>> expected_shapes);
|
| /external/tensorflow/tensorflow/contrib/data/python/kernel_tests/ |
| D | assert_element_shape_test.py | 44 expected_shapes = (tensor_shape.TensorShape(2),
46 self.assertEqual(expected_shapes,
49 result = dataset.apply(batching.assert_element_shape(expected_shapes))
50 self.assertEqual(expected_shapes,
91 expected_shapes = (tensor_shape.TensorShape(2),
93 result = dataset.apply(batching.assert_element_shape(expected_shapes))
94 self.assertEqual(expected_shapes,
190 expected_shapes = (tensor_shape.TensorShape(2),
192 result = dataset.apply(batching.assert_element_shape(expected_shapes))
193 self.assertEqual(expected_shapes,
|
| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| D | batching.py | 177 def assert_element_shape(expected_shapes): argument
200 def _merge_output_shapes(original_shapes, expected_shapes): argument
202 flat_new_shapes = nest.flatten_up_to(original_shapes, expected_shapes)
211 flatten_shapes = nest.flatten(expected_shapes)
220 dataset_ops.get_legacy_output_shapes(dataset), expected_shapes)
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | choose_fastest_dataset_test.py | 41 expected_shapes=dataset_ops.get_legacy_output_shapes(dataset))
49 expected_shapes=dataset_ops.get_legacy_output_shapes(dataset))
62 expected_shapes=dataset_ops.get_legacy_output_shapes(dataset_a))
|
| D | choose_fastest_branch_dataset_test.py | 51 expected_shapes=dataset.output_shapes)
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | np_utils_test.py | 32 expected_shapes = [(1, num_classes),
42 expected_shapes):
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | group_by_reducer_test.py | 49 expected_shapes=tensor_shape.scalar(),
68 expected_shapes=tensor_shape.scalar(),
84 expected_shapes=tensor_shape.scalar(),
103 expected_shapes=tensor_shape.scalar(),
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | range_dataset_op_test.cc | 180 std::vector<PartialTensorShape> expected_shapes({PartialTensorShape({})}); in TEST_F() local
181 EXPECT_EQ(range_dataset->output_shapes().size(), expected_shapes.size()); in TEST_F()
184 range_dataset->output_shapes()[i].IsIdenticalTo(expected_shapes[i])); in TEST_F()
308 std::vector<PartialTensorShape> expected_shapes({PartialTensorShape({})}); in TEST_F() local
309 EXPECT_EQ(iterator->output_shapes().size(), expected_shapes.size()); in TEST_F()
311 EXPECT_TRUE(iterator->output_shapes()[i].IsIdenticalTo(expected_shapes[i])); in TEST_F()
|
| D | map_dataset_op_test.cc | 248 std::vector<PartialTensorShape> expected_shapes({PartialTensorShape({})}); in TEST_F() local
249 EXPECT_EQ(map_dataset->output_shapes().size(), expected_shapes.size()); in TEST_F()
252 map_dataset->output_shapes()[i].IsIdenticalTo(expected_shapes[i])); in TEST_F()
403 std::vector<PartialTensorShape> expected_shapes({PartialTensorShape({})}); in TEST_F() local
404 EXPECT_EQ(iterator->output_shapes().size(), expected_shapes.size()); in TEST_F()
406 EXPECT_TRUE(iterator->output_shapes()[i].IsIdenticalTo(expected_shapes[i])); in TEST_F()
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | test_base.py | 99 expected_shapes=None, argument
144 if expected_shapes:
145 self.assertEqual(expected_shapes,
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | indexed_dataset_op.cc | 377 auto expected_shapes = materialized_resource->output_shapes(); in Compute() local
381 ctx, expected_shapes[i].IsCompatibleWith(out_tensors[i].shape()), in Compute()
385 expected_shapes[i], ", got: ", out_tensors[i].shape(), ")")); in Compute()
|
| /external/tensorflow/tensorflow/contrib/meta_graph_transform/ |
| D | meta_graph_transform_test.py | 186 expected_shapes = [shapes[0], shapes[2]]
210 expected_shapes)
213 len(expected_shapes))
216 len(expected_shapes))
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | structure_test.py | 65 expected_shapes): argument
70 for expected, actual in zip(expected_shapes, s._flat_shapes):
|