Searched refs:ShapeToTensorProperty (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils_test.cc | 56 OpInfo::TensorProperties ShapeToTensorProperty(const std::vector<int>& shapes, in ShapeToTensorProperty() function
184 CalculateTensorSize(ShapeToTensorProperty({1}, DT_FLOAT))); in TEST()
186 CalculateTensorSize(ShapeToTensorProperty({4, 4}, DT_FLOAT))); in TEST()
188 CalculateTensorSize(ShapeToTensorProperty({10, 10, 10}, DT_HALF))); in TEST()
191 CalculateTensorSize(ShapeToTensorProperty({100, 7, 8, 99}, DT_FLOAT))); in TEST()
194 OpInfo::TensorProperties t = ShapeToTensorProperty({100, 7, 8, 99}, DT_FLOAT); in TEST()
201 CalculateTensorSize(ShapeToTensorProperty({-1, 7, 8, 99}, DT_FLOAT))); in TEST()
204 CalculateTensorSize(ShapeToTensorProperty({-1, 7, -1, 99}, DT_FLOAT))); in TEST()
210 ShapeToTensorProperty({4, 4}, DT_FLOAT), // 0 in TEST()
211 ShapeToTensorProperty({-1, 7, -1, 99}, DT_FLOAT) // 1 in TEST()
|