Searched refs:ValidateOpDimensionsFromInputs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 540 void ValidateOpDimensionsFromInputs(const int n, const int h, const int w, in ValidateOpDimensionsFromInputs() function in tensorflow::grappler::OpLevelCostEstimatorTest
1378 ValidateOpDimensionsFromInputs(10, 20, 20, 100, 3, 3, 2, 2, f, p); in TEST_F()
1379 ValidateOpDimensionsFromInputs(10, 20, 20, 100, 1, 1, 3, 3, f, p); in TEST_F()
1380 ValidateOpDimensionsFromInputs(10, 200, 200, 100, 5, 5, 3, 3, f, p); in TEST_F()
1381 ValidateOpDimensionsFromInputs(10, 14, 14, 3840, 3, 3, 2, 2, f, p); in TEST_F()
|