Searched refs:MakeTensorShapeFromDimensions (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | generic_layout_optimizer_transposer_test.cc | 540 TensorShapeProto MakeTensorShapeFromDimensions(absl::Span<const int> dims) { in MakeTensorShapeFromDimensions() function
565 TensorShapeProto input_shape = MakeTensorShapeFromDimensions({1, 2, 3, 4}); in TEST_F()
566 TensorShapeProto expected_shape = MakeTensorShapeFromDimensions({1, 4, 2, 3}); in TEST_F()
731 MakeTensorShapeFromDimensions({32, 5, 3, 16}); in TEST_F()
733 MakeTensorShapeFromDimensions({32, 16, 5, 3}); in TEST_F()
3790 TensorShapeProto input_shape = MakeTensorShapeFromDimensions({1, 2, 3, 4}); in TEST()
3791 TensorShapeProto expected_shape = MakeTensorShapeFromDimensions({1, 4, 2, 3}); in TEST()
3801 MakeTensorShapeFromDimensions({1, 2, 3, 4, 5, 6, 7, 8}); in TEST()
3803 MakeTensorShapeFromDimensions({1, 2, 7, 8, 3, 4, 5, 6}); in TEST()
3811 MakeTensorShapeFromDimensions({1, 2, 3, 4, 5, 6, 7, 8}); in TEST()
[all …]
|