Searched refs:sqrt_op (Results 1 – 2 of 2) sorted by relevance
45 const auto* sqrt_op = sqrt_it->get(); in Run() local46 if (sqrt_op->type != OperatorType::kSqrt) { in Run()50 CHECK_EQ(sqrt_op->inputs.size(), 1); in Run()51 CHECK_EQ(sqrt_op->outputs.size(), 1); in Run()56 Operator* prev_to_sqrt_op = GetOpWithOutput(*model, sqrt_op->inputs[0]); in Run()89 l2pool_op->outputs = sqrt_op->outputs; in Run()105 model->EraseArray(sqrt_op->inputs[0]); in Run()110 model->operators.erase(FindOperator(model, sqrt_op)); in Run()
1099 tensorflow::NodeDef* sqrt_op = tensorflow_graph->add_node(); in ConvertL2PoolOperator() local1100 sqrt_op->set_op("Sqrt"); in ConvertL2PoolOperator()1101 sqrt_op->set_name(src_op.outputs[0]); in ConvertL2PoolOperator()1102 *sqrt_op->add_input() = avgpool_output; in ConvertL2PoolOperator()1103 (*sqrt_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertL2PoolOperator()1118 tensorflow::NodeDef* sqrt_op = tensorflow_graph->add_node(); in ConvertSqrtOperator() local1119 sqrt_op->set_op("Sqrt"); in ConvertSqrtOperator()1120 sqrt_op->set_name(src_op.outputs[0]); in ConvertSqrtOperator()1122 *sqrt_op->add_input() = src_op.inputs[0]; in ConvertSqrtOperator()1123 (*sqrt_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertSqrtOperator()