Searched refs:avgpool_op (Results 1 – 2 of 2) sorted by relevance
48 auto avgpool_op = new AveragePoolOperator; in RunIdentifyL2Pool() local49 avgpool_op->inputs = {sq_op->outputs[0]}; in RunIdentifyL2Pool()50 avgpool_op->outputs = {"Avgpooloutput"}; in RunIdentifyL2Pool()56 sqrt_op->inputs = {avgpool_op->outputs[0]}; in RunIdentifyL2Pool()61 model.operators.push_back(std::unique_ptr<Operator>(avgpool_op)); in RunIdentifyL2Pool()
1005 tensorflow::NodeDef* avgpool_op = tensorflow_graph->add_node(); in ConvertAveragePoolOperator() local1006 avgpool_op->set_op("AvgPool"); in ConvertAveragePoolOperator()1007 avgpool_op->set_name(src_op.outputs[0]); in ConvertAveragePoolOperator()1008 *avgpool_op->add_input() = src_op.inputs[0]; in ConvertAveragePoolOperator()1009 auto& strides = (*avgpool_op->mutable_attr())["strides"]; in ConvertAveragePoolOperator()1022 (*avgpool_op->mutable_attr())["padding"].set_s(padding); in ConvertAveragePoolOperator()1023 (*avgpool_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertAveragePoolOperator()1024 auto& ksize = (*avgpool_op->mutable_attr())["ksize"]; in ConvertAveragePoolOperator()1089 tensorflow::NodeDef* avgpool_op = tensorflow_graph->add_node(); in ConvertL2PoolOperator() local1090 avgpool_op->set_op("AvgPool"); in ConvertL2PoolOperator()[all …]