Searched refs:avgpool_op (Results 1 – 1 of 1) sorted by relevance
997 tensorflow::NodeDef* avgpool_op = tensorflow_graph->add_node(); in ConvertAveragePoolOperator() local998 avgpool_op->set_op("AvgPool"); in ConvertAveragePoolOperator()999 avgpool_op->set_name(src_op.outputs[0]); in ConvertAveragePoolOperator()1000 *avgpool_op->add_input() = src_op.inputs[0]; in ConvertAveragePoolOperator()1001 auto& strides = (*avgpool_op->mutable_attr())["strides"]; in ConvertAveragePoolOperator()1014 (*avgpool_op->mutable_attr())["padding"].set_s(padding); in ConvertAveragePoolOperator()1015 (*avgpool_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertAveragePoolOperator()1016 auto& ksize = (*avgpool_op->mutable_attr())["ksize"]; in ConvertAveragePoolOperator()1081 tensorflow::NodeDef* avgpool_op = tensorflow_graph->add_node(); in ConvertL2PoolOperator() local1082 avgpool_op->set_op("AvgPool"); in ConvertL2PoolOperator()[all …]