Searched refs:square_op (Results 1 – 3 of 3) sorted by relevance
54 const Operator* square_op; in Run() local75 square_op = GetOpWithOutput(*model, avpool_op->inputs[0]); in Run()76 CHECK_EQ(square_op->inputs.size(), 1); in Run()77 if (square_op->type != OperatorType::kSquare) { in Run()81 LogName(*square_op)); in Run()88 l2pool_op->inputs = {square_op->inputs[0]}; in Run()108 model->operators.erase(FindOperator(model, square_op)); in Run()
125 Operator* square_op = GetOpWithOutput(*model, sum_op->inputs[0]); in Run() local126 if (square_op->type != OperatorType::kSquare) { in Run()130 LogName(*square_op)); in Run()134 CHECK_EQ(square_op->inputs.size(), 1); in Run()136 if (square_op->inputs[0] != div_or_mul_op->inputs[0]) { in Run()140 LogName(*square_op)); in Run()153 model->operators.erase(FindOperator(model, square_op)); in Run()
909 tensorflow::NodeDef* square_op = tensorflow_graph->add_node(); in ConvertL2NormalizationOperator() local910 square_op->set_op("Square"); in ConvertL2NormalizationOperator()911 square_op->set_name(square_output); in ConvertL2NormalizationOperator()912 *square_op->add_input() = src_op.inputs[0]; in ConvertL2NormalizationOperator()913 (*square_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertL2NormalizationOperator()1066 tensorflow::NodeDef* square_op = tensorflow_graph->add_node(); in ConvertL2PoolOperator() local1067 square_op->set_op("Square"); in ConvertL2PoolOperator()1068 square_op->set_name(square_output); in ConvertL2PoolOperator()1069 *square_op->add_input() = src_op.inputs[0]; in ConvertL2PoolOperator()1070 (*square_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertL2PoolOperator()[all …]