Home
last modified time | relevance | path

Searched refs:maxpool_op (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc969 tensorflow::NodeDef* maxpool_op = tensorflow_graph->add_node(); in ConvertMaxPoolOperator() local
970 maxpool_op->set_op("MaxPool"); in ConvertMaxPoolOperator()
971 maxpool_op->set_name(src_op.outputs[0]); in ConvertMaxPoolOperator()
972 *maxpool_op->add_input() = src_op.inputs[0]; in ConvertMaxPoolOperator()
973 auto& strides = (*maxpool_op->mutable_attr())["strides"]; in ConvertMaxPoolOperator()
986 (*maxpool_op->mutable_attr())["padding"].set_s(padding); in ConvertMaxPoolOperator()
987 (*maxpool_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertMaxPoolOperator()
988 auto& ksize = (*maxpool_op->mutable_attr())["ksize"]; in ConvertMaxPoolOperator()