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

  /external/tensorflow/tensorflow/tools/graph_transforms/python/
transform_graph_test.py 64 relu_op = input_graph_def.node.add()
65 relu_op.op = "Relu"
66 relu_op.attr["T"].CopyFrom(attr_value_pb2.AttrValue(
68 relu_op.name = "relu_op"
69 relu_op.input.extend(["add_op"])
  /external/tensorflow/tensorflow/contrib/lite/toco/
export_tensorflow.cc 605 auto* relu_op = tensorflow_graph->add_node(); local
606 relu_op->set_op("Relu");
607 relu_op->set_name(src_op.outputs[0]);
608 *relu_op->add_input() = src_op.inputs[0];
609 (*relu_op->mutable_attr())["T"].set_type(DT_FLOAT);
653 auto* relu_op = tensorflow_graph->add_node(); local
654 relu_op->set_op("Relu6");
655 relu_op->set_name(src_op.outputs[0]);
656 *relu_op->add_input() = src_op.inputs[0];
657 (*relu_op->mutable_attr())["T"].set_type(DT_FLOAT)
662 auto* relu_op = tensorflow_graph->add_node(); local
    [all...]
  /external/tensorflow/tensorflow/tools/graph_transforms/
quantize_nodes_test.cc 404 Output relu_op = Relu(root.WithOpName("relu_op"), constant_op); local
409 TestQuantizedVersusFloatGraph(float_graph_def, {}, {"relu_op"});
767 Output relu_op = Relu(root.WithOpName("relu_op"), dequantize_op); local
791 Output relu_op = Relu(root.WithOpName("relu_op"), placeholder_op); local
1093 Output relu_op = Relu(root.WithOpName("relu_op"), bias_add_op); local
    [all...]

Completed in 62 milliseconds