Searched refs:DescribePoolingOp (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 428 OpContext DescribePoolingOp(const string& op_name, const std::vector<int>& x, in DescribePoolingOp() function
470 OpContext op_context = DescribePoolingOp("MaxPool", x, {1, 1, 1, 1}, in DescribeFusedBatchNorm()
549 op_context = DescribePoolingOp("MaxPool", {n, h, w, c}, {1, kx, ky, 1}, in ValidateOpDimensionsFromInputs()
554 op_context = DescribePoolingOp("MaxPool", {n, c, h, w}, {1, 1, kx, ky}, in ValidateOpDimensionsFromInputs()
1390 OpContext op_context = DescribePoolingOp( in TEST_F()
1434 DescribePoolingOp("MaxPoolGrad", {n, in, in, c}, {1, k, k, 1}, in TEST_F()
1477 OpContext op_context = DescribePoolingOp( in TEST_F()
1521 DescribePoolingOp("AvgPoolGrad", {n, in, in, c}, {1, k, k, 1}, in TEST_F()
|