Searched refs:CreatePoolOp (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 5026 NodeDef CreatePoolOp(DataType tf_type, std::vector<int> ksize, in CreatePoolOp() function
5056 return CreatePoolOp<ops::MaxPool3D>(tf_type, ksize, strides, padding, in TEST_P()
5059 return CreatePoolOp<ops::MaxPool>(tf_type, ksize, strides, padding, in TEST_P()
5064 return CreatePoolOp<ops::AvgPool3D>(tf_type, ksize, strides, padding, in TEST_P()
5067 return CreatePoolOp<ops::AvgPool>(tf_type, ksize, strides, padding, in TEST_P()
|