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

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
identify_l2_pool.cc 52 const Operator* square_op; local
66 square_op = GetOpWithOutput(*model, avpool_op->inputs[0]);
67 CHECK_EQ(square_op->inputs.size(), 1);
68 if (square_op->type != OperatorType::kTensorFlowSquare) {
72 LogName(*square_op));
79 l2pool_op->inputs = {square_op->inputs[0]};
99 model->operators.erase(FindOperator(model, square_op));
identify_l2_normalization.cc 124 Operator* square_op = GetOpWithOutput(*model, sum_op->inputs[0]); local
125 if (square_op->type != OperatorType::kTensorFlowSquare) {
129 LogName(*square_op));
133 CHECK_EQ(square_op->inputs.size(), 1);
135 if (square_op->inputs[0] != div_or_mul_op->inputs[0]) {
139 LogName(*square_op));
152 model->operators.erase(FindOperator(model, square_op));
  /external/tensorflow/tensorflow/contrib/lite/toco/
export_tensorflow.cc 779 auto* square_op = tensorflow_graph->add_node(); local
780 square_op->set_op("Square");
781 square_op->set_name(square_output);
782 *square_op->add_input() = src_op.inputs[0];
783 (*square_op->mutable_attr())["T"].set_type(DT_FLOAT);
930 auto* square_op = tensorflow_graph->add_node(); local
972 auto* square_op = tensorflow_graph->add_node(); local
    [all...]

Completed in 50 milliseconds