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

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
identify_relu1.cc 79 const auto* max_op = local
83 CHECK_EQ(max_op->inputs.size(), 2);
84 if (min_op->outputs.size() != 1 || max_op->outputs.size() != 1) {
92 GetSingleScalarInputIndexOfBinaryOp(model, max_op, -1.0f);
  /external/tensorflow/tensorflow/contrib/lite/toco/
export_tensorflow.cc 636 auto* max_op = tensorflow_graph->add_node(); local
637 max_op->set_op("Maximum");
638 max_op->set_name(max_output);
639 *max_op->add_input() = src_op.inputs[0];
640 *max_op->add_input() = max_bounds;
641 (*max_op->mutable_attr())["T"].set_type(DT_FLOAT);
    [all...]

Completed in 67 milliseconds