| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| H A D | drop_im2col_arrays.cc | 27 auto* conv_op = static_cast<ConvOperator*>(conv_it->get()); local
28 if (conv_op->outputs.size() < 2) {
34 CHECK_EQ(conv_op->outputs.size(), 2);
35 model->EraseArray(conv_op->outputs[1]);
36 conv_op->outputs.resize(1);
37 AddMessageF("Dropped an im2col array for %s", LogName(*conv_op));
|
| H A D | convert_pure_conv_to_depthwise.cc | 32 const auto* conv_op = static_cast<ConvOperator*>(conv_it->get()); local
33 if (conv_op->stride_width != conv_op->stride_height) {
36 auto& weights_array = model->GetArray(conv_op->inputs[1]);
53 LogName(*conv_op));
56 depthwiseconv_op->inputs = conv_op->inputs;
58 depthwiseconv_op->outputs = {conv_op->outputs[0]};
59 if (conv_op->outputs.size() > 1) {
61 model->EraseArray(conv_op->outputs[1]);
64 conv_op
[all...] |
| H A D | create_im2col_arrays.cc | 33 auto* conv_op = static_cast<ConvOperator*>(conv_it->get()); local
34 if (conv_op->outputs.size() == 2) {
38 const auto& weights_array = model->GetArray(conv_op->inputs[1]);
48 if (kwidth == 1 && kheight == 1 && conv_op->stride_width == 1 &&
49 conv_op->stride_height == 1) {
55 CHECK_EQ(conv_op->outputs.size(), 1);
57 AvailableArrayName(*model, conv_op->inputs[0] + "_im2col");
59 conv_op->outputs.push_back(im2col_array_name);
63 LogName(*conv_op), kwidth, kheight, conv_op
[all...] |
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | round_weights_test.cc | 59 Output conv_op = Conv2D(root.WithOpName("output"), input_op, weights_op, local
|
| H A D | flatten_atrous_test.cc | 70 Output conv_op = TConvOp(root.WithOpName("conv_op"), space_to_batch_op, local
79 root.WithOpName("output"), conv_op, block_shape_op, crops_op);
106 EXPECT_EQ(conv_op.node()->type_string(), modified_graph_def.node(2).op());
|
| H A D | fold_batch_norms_test.cc | 56 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, local
64 Output mul_op = Mul(root.WithOpName("output"), conv_op, mul_values_op);
107 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, local
115 Output mul_op = Mul(root.WithOpName("output"), conv_op, mul_values_op);
123 Mul(root.WithOpName("output_2"), conv_op, mul_values_op_2);
|
| H A D | fuse_convolutions_test.cc | 72 Output conv_op = Conv2D(root.WithOpName("output"), pad_op, weights_op, local
122 Output conv_op = Conv2D(root.WithOpName("output"), resize_op, weights_op, local
174 Output conv_op = Conv2D(root.WithOpName("output"), pad_op, weights_op, local
|
| H A D | quantize_weights_test.cc | 55 Output conv_op = ops::Conv2D(root.WithOpName("output"), input_op, local
|
| H A D | fold_old_batch_norms_test.cc | 56 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, local
91 AddNodeInput("conv_op", &batch_norm_node);
140 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, local
169 AddNodeInput("conv_op", &batch_norm_node);
|
| H A D | quantize_nodes_test.cc | 335 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, filter_op, local
341 TestQuantizedVersusFloatGraph(float_graph_def, {}, {"conv_op"});
|
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| H A D | dump_graphviz.cc | 245 const auto& conv_op = static_cast<const ConvOperator&>(op); local
247 AppendF(&node_properties.label, "\\n%dx%d/%s", conv_op.stride_width,
248 conv_op.stride_height,
249 conv_op.padding.type == PaddingType::kSame ? "S" : "V");
253 const auto& conv_op = static_cast<const DepthwiseConvOperator&>(op); local
255 AppendF(&node_properties.label, "\\n%dx%d/%s", conv_op.stride_width,
256 conv_op.stride_height,
257 conv_op.padding.type == PaddingType::kSame ? "S" : "V");
|