HomeSort by relevance Sort by last modified time
    Searched refs:square_op (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/lite/toco/graph_transformations/
identify_l2_pool.cc 54 const Operator* square_op; local
75 square_op = GetOpWithOutput(*model, avpool_op->inputs[0]);
76 CHECK_EQ(square_op->inputs.size(), 1);
77 if (square_op->type != OperatorType::kSquare) {
81 LogName(*square_op));
88 l2pool_op->inputs = {square_op->inputs[0]};
108 model->operators.erase(FindOperator(model, square_op));
identify_l2_normalization.cc 125 Operator* square_op = GetOpWithOutput(*model, sum_op->inputs[0]); local
126 if (square_op->type != OperatorType::kSquare) {
130 LogName(*square_op));
134 CHECK_EQ(square_op->inputs.size(), 1);
136 if (square_op->inputs[0] != div_or_mul_op->inputs[0]) {
140 LogName(*square_op));
153 model->operators.erase(FindOperator(model, square_op));
  /external/tensorflow/tensorflow/lite/toco/
export_tensorflow.cc 909 tensorflow::NodeDef* square_op = tensorflow_graph->add_node(); local
1066 tensorflow::NodeDef* square_op = tensorflow_graph->add_node(); local
1108 tensorflow::NodeDef* square_op = tensorflow_graph->add_node(); local
    [all...]

Completed in 2256 milliseconds