Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Didentify_l2_pool.cc40 const Operator* square_op; in Run() local
61 square_op = GetOpWithOutput(*model, avpool_op->inputs[0]); in Run()
62 CHECK_EQ(square_op->inputs.size(), 1); in Run()
63 if (square_op->type != OperatorType::kSquare) { in Run()
67 LogName(*square_op)); in Run()
74 l2pool_op->inputs = {square_op->inputs[0]}; in Run()
89 DeleteOpAndArrays(model, square_op); in Run()
Didentify_l2_normalization.cc111 Operator* square_op = GetOpWithOutput(*model, sum_op->inputs[0]); in Run() local
112 if (square_op->type != OperatorType::kSquare) { in Run()
116 LogName(*square_op)); in Run()
120 CHECK_EQ(square_op->inputs.size(), 1); in Run()
122 if (square_op->inputs[0] != div_or_mul_op->inputs[0]) { in Run()
126 LogName(*square_op)); in Run()
139 DeleteOpAndArrays(model, square_op); in Run()
/external/XNNPACK/test/
Dsquare-operator-tester.h105 xnn_operator_t square_op = nullptr; in TestF32() local
110 0, &square_op)); in TestF32()
111 ASSERT_NE(nullptr, square_op); in TestF32()
114 …std::unique_ptr<xnn_operator, decltype(&xnn_delete_operator)> auto_square_op(square_op, xnn_delete… in TestF32()
118 square_op, in TestF32()
124 xnn_run_operator(square_op, nullptr /* thread pool */)); in TestF32()
/external/XNNPACK/src/operators/
Dunary-elementwise-nc.c755 xnn_operator_t square_op, in xnn_setup_square_nc_f32() argument
761 if (square_op->type != xnn_operator_type_square_nc_f32) { in xnn_setup_square_nc_f32()
764 xnn_operator_type_to_string(square_op->type)); in xnn_setup_square_nc_f32()
767 square_op->state = xnn_run_state_invalid; in xnn_setup_square_nc_f32()
770 square_op, in xnn_setup_square_nc_f32()
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc917 tensorflow::NodeDef* square_op = tensorflow_graph->add_node(); in ConvertL2NormalizationOperator() local
918 square_op->set_op("Square"); in ConvertL2NormalizationOperator()
919 square_op->set_name(square_output); in ConvertL2NormalizationOperator()
920 *square_op->add_input() = src_op.inputs[0]; in ConvertL2NormalizationOperator()
921 (*square_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertL2NormalizationOperator()
1074 tensorflow::NodeDef* square_op = tensorflow_graph->add_node(); in ConvertL2PoolOperator() local
1075 square_op->set_op("Square"); in ConvertL2PoolOperator()
1076 square_op->set_name(square_output); in ConvertL2PoolOperator()
1077 *square_op->add_input() = src_op.inputs[0]; in ConvertL2PoolOperator()
1078 (*square_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertL2PoolOperator()
[all …]
/external/XNNPACK/include/
Dxnnpack.h1579 xnn_operator_t square_op,