| /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 | 42 std::map<string, std::vector<PartialTensorShape>> expected_shapes);
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | rebatch_dataset_test.py | 168 expected_shapes = [[None]]
169 self.assertEqual(expected_shapes, _flat_shapes(rebatched_dataset))
178 expected_shapes = [[2]]
179 self.assertEqual(expected_shapes, _flat_shapes(rebatched_dataset))
187 expected_shapes = [[None]]
188 self.assertEqual(expected_shapes, _flat_shapes(rebatched_dataset))
196 expected_shapes = [[2]]
197 self.assertEqual(expected_shapes, _flat_shapes(rebatched_dataset))
205 expected_shapes = [[None]]
206 self.assertEqual(expected_shapes, _flat_shapes(rebatched_dataset))
[all …]
|
| D | group_by_reducer_test.py | 47 expected_shapes=tensor_shape.TensorShape([]),
67 expected_shapes=tensor_shape.TensorShape([]),
84 expected_shapes=tensor_shape.TensorShape([]),
104 expected_shapes=tensor_shape.TensorShape([]),
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | reduce_decomposer.cc | 80 std::vector<Shape> expected_shapes(reduce->input_count()); in HandleReduce() local
82 expected_shapes[i] = ExpectedOutputShape(reduce, i); in HandleReduce()
98 if (ShapeUtil::MakeMaybeTupleShape(expected_shapes) != in HandleReduce()
105 ShapeUtil::MakeMaybeTupleShape(expected_shapes)); in HandleReduce()
|
| D | hlo_creation_utils.cc | 459 std::vector<Shape> expected_shapes; in MakeReduceHlo() local
461 expected_shapes.push_back(ShapeUtil::FilterDimensions( in MakeReduceHlo()
468 auto output_shape = ShapeUtil::MakeMaybeTupleShape(expected_shapes); in MakeReduceHlo()
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | from_tensors_test.py | 178 expected_shapes=[[[None, 3], [None, 3], [None, 2], [None, 2]]]) +
185 expected_shapes=[[[1, 3], [1, 3], [1, 2], [1, 2]]]))
186 def testNestedStructure(self, components, expected_shapes): argument
200 self.assertEqual(expected_shapes, [
213 self.assertEqual(expected_shapes, [
|
| D | test_base.py | 169 expected_shapes=None, argument
214 if expected_shapes:
215 self.assertEqual(expected_shapes,
|
| /external/pytorch/test/ |
| D | test_meta.py | 1383 def _norm_backwards_test_helper(self, op, args, output_mask, expected_shapes): argument
1394 assertEqualShapes(grads[0], expected_shapes[0])
1395 assertEqualShapes(grads[1], expected_shapes[1])
1396 assertEqualShapes(grads[2], expected_shapes[2])
1409 assertEqualShapes(out_kwargs["out0"], expected_shapes[0])
1410 assertEqualShapes(out_kwargs["out1"], expected_shapes[1])
1411 assertEqualShapes(out_kwargs["out2"], expected_shapes[2])
1436 expected_shapes = (
1444 args, output_mask, expected_shapes)
1468 expected_shapes = (
[all …]
|
| D | test_prims.py | 131 expected_shapes = [(2, 2, 2), (4, 2), (2, 4), (8,)]
133 for (start, end), shape in zip(dim_ranges, expected_shapes):
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | tf_trt_integration_test_base.py | 386 for expected_shapes, current_input_data in zip(
391 self.assertEqual(len(expected_shapes), len(new_val))
392 for expected_shape, actual_val in zip(expected_shapes, new_val):
409 for expected_shapes, current_input_data in zip(params.expected_output_dims,
426 self.assertEqual(len(expected_shapes), len(new_val))
427 for expected_shape, actual_val in zip(expected_shapes, new_val):
|
| /external/pytorch/test/torch_np/numpy_tests/core/ |
| D | test_shape_base.py | 495 expected_shapes = [(10, 3), (3, 10), (3, 10), (10, 3)]
496 for axis, expected_shape in zip(axes, expected_shapes):
505 expected_shapes = [
513 for axis, expected_shape in zip(axes, expected_shapes):
|
| /external/pytorch/test/cpp/c10d/ |
| D | ProcessGroupGlooTest.cpp | 234 std::vector<std::vector<int64_t>> expected_shapes, in checkProfiledEvents() argument
237 EXPECT_EQ(expected_count, expected_shapes.size()); in checkProfiledEvents()
247 if (shapesVec[0] == expected_shapes[i]) { in checkProfiledEvents()
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | structure_test.py | 508 expected_shapes = expected_shapes_fn()
514 self.assertLen(flat_shapes, len(expected_shapes))
515 for expected, actual in zip(expected_shapes, flat_shapes):
|
| /external/pytorch/torch/testing/_internal/distributed/ |
| D | distributed_test.py | 1635 expected_shapes = [
1639 self.assertTrue(event.input_shapes in expected_shapes)
1706 expected_shapes = [
1711 self.assertTrue(event.input_shapes in expected_shapes)
1937 expected_shapes = {
1945 event.input_shapes in expected_shapes.values()
1948 self.assertEqual(event.input_shapes, expected_shapes[rank])
|