Home
last modified time | relevance | path

Searched refs:sqrt_op (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Didentify_l2_pool.cc31 const auto* sqrt_op = sqrt_it->get(); in Run() local
32 if (sqrt_op->type != OperatorType::kSqrt) { in Run()
36 CHECK_EQ(sqrt_op->inputs.size(), 1); in Run()
37 CHECK_EQ(sqrt_op->outputs.size(), 1); in Run()
42 Operator* prev_to_sqrt_op = GetOpWithOutput(*model, sqrt_op->inputs[0]); in Run()
75 l2pool_op->outputs = sqrt_op->outputs; in Run()
91 DeleteOpAndArrays(model, sqrt_op); in Run()
/external/XNNPACK/test/
Dsquare-root-operator-tester.h104 xnn_operator_t sqrt_op = nullptr; in TestF32() local
109 0, &sqrt_op)); in TestF32()
110 ASSERT_NE(nullptr, sqrt_op); in TestF32()
113 …std::unique_ptr<xnn_operator, decltype(&xnn_delete_operator)> auto_sqrt_op(sqrt_op, xnn_delete_ope… in TestF32()
117 sqrt_op, in TestF32()
123 xnn_run_operator(sqrt_op, nullptr /* thread pool */)); in TestF32()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/
Didentify_l2_pool_test.cc55 auto sqrt_op = new TensorFlowSqrtOperator; in RunIdentifyL2Pool() local
56 sqrt_op->inputs = {avgpool_op->outputs[0]}; in RunIdentifyL2Pool()
57 sqrt_op->outputs = {"Sqrtoutput"}; in RunIdentifyL2Pool()
60 model.operators.push_back(std::unique_ptr<Operator>(sqrt_op)); in RunIdentifyL2Pool()
Didentify_l2_normalization_test.cc57 auto sqrt_op = new TensorFlowSqrtOperator; in RunIdentifyL2Normalization() local
58 sqrt_op->inputs = {sum_op->outputs[0]}; in RunIdentifyL2Normalization()
59 sqrt_op->outputs = {"squarertoutput"}; in RunIdentifyL2Normalization()
65 div_op->inputs = {"input0", sqrt_op->outputs[0]}; in RunIdentifyL2Normalization()
70 model.operators.push_back(std::unique_ptr<Operator>(sqrt_op)); in RunIdentifyL2Normalization()
/external/XNNPACK/bench/
Dsquare-root.cc47 xnn_operator_t sqrt_op = nullptr; in xnnpack_square_root_f32() local
50 0 /* flags */, &sqrt_op); in xnnpack_square_root_f32()
51 if (status != xnn_status_success || sqrt_op == nullptr) { in xnnpack_square_root_f32()
57 sqrt_op, in xnnpack_square_root_f32()
67 status = xnn_run_operator(sqrt_op, nullptr /* thread pool */); in xnnpack_square_root_f32()
74 status = xnn_delete_operator(sqrt_op); in xnnpack_square_root_f32()
/external/tensorflow/tensorflow/c/experimental/ops/
Dmath_ops.cc173 AbstractOperationPtr sqrt_op(ctx->CreateOperation()); in Sqrt() local
174 TF_RETURN_IF_ERROR(sqrt_op->Reset("Sqrt", /*raw_device_name=*/nullptr)); in Sqrt()
175 TF_RETURN_IF_ERROR(MaybeSetOpName(sqrt_op.get(), name)); in Sqrt()
176 TF_RETURN_IF_ERROR(sqrt_op->AddInput(inputs[0])); in Sqrt()
179 Status s = sqrt_op->Execute(outputs, &num_retvals); in Sqrt()
/external/XNNPACK/src/operators/
Dunary-elementwise-nc.c777 xnn_operator_t sqrt_op, in xnn_setup_square_root_nc_f32() argument
783 if (sqrt_op->type != xnn_operator_type_square_root_nc_f32) { in xnn_setup_square_root_nc_f32()
786 xnn_operator_type_to_string(sqrt_op->type)); in xnn_setup_square_root_nc_f32()
789 sqrt_op->state = xnn_run_state_invalid; in xnn_setup_square_root_nc_f32()
792 sqrt_op, in xnn_setup_square_root_nc_f32()
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc1107 tensorflow::NodeDef* sqrt_op = tensorflow_graph->add_node(); in ConvertL2PoolOperator() local
1108 sqrt_op->set_op("Sqrt"); in ConvertL2PoolOperator()
1109 sqrt_op->set_name(src_op.outputs[0]); in ConvertL2PoolOperator()
1110 *sqrt_op->add_input() = avgpool_output; in ConvertL2PoolOperator()
1111 (*sqrt_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertL2PoolOperator()
1126 tensorflow::NodeDef* sqrt_op = tensorflow_graph->add_node(); in ConvertSqrtOperator() local
1127 sqrt_op->set_op("Sqrt"); in ConvertSqrtOperator()
1128 sqrt_op->set_name(src_op.outputs[0]); in ConvertSqrtOperator()
1130 *sqrt_op->add_input() = src_op.inputs[0]; in ConvertSqrtOperator()
1131 (*sqrt_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertSqrtOperator()
/external/XNNPACK/include/
Dxnnpack.h1593 xnn_operator_t sqrt_op,