Home
last modified time | relevance | path

Searched refs:fused_conv (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dfuse_convolutions.cc60 NodeDef fused_conv; in FuseResizePadAndConv() local
61 fused_conv.set_op("FusedResizeAndPadConv2D"); in FuseResizePadAndConv()
62 fused_conv.set_name(match.node.name()); in FuseResizePadAndConv()
63 AddNodeInput(resize_node.input(0), &fused_conv); in FuseResizePadAndConv()
64 AddNodeInput(resize_node.input(1), &fused_conv); in FuseResizePadAndConv()
65 AddNodeInput(mirror_pad_node.input(1), &fused_conv); in FuseResizePadAndConv()
66 AddNodeInput(conv_node.input(1), &fused_conv); in FuseResizePadAndConv()
68 &fused_conv); in FuseResizePadAndConv()
69 CopyNodeAttr(mirror_pad_node, "mode", "mode", &fused_conv); in FuseResizePadAndConv()
70 CopyNodeAttr(conv_node, "T", "T", &fused_conv); in FuseResizePadAndConv()
[all …]
DREADME.md990 NodeDef fused_conv;
991 fused_conv.set_op("FusedResizeAndPadConv2D");
992 fused_conv.set_name(match.node.name());
993 AddNodeInput(resize_node.input(0), &fused_conv);
994 AddNodeInput(resize_node.input(1), &fused_conv);
995 AddNodeInput(pad_dims_node.name(), &fused_conv);
996 AddNodeInput(conv_node.input(1), &fused_conv);
998 &fused_conv);
999 SetNodeAttr("mode", "REFLECT", &fused_conv);
1000 CopyNodeAttr(conv_node, "T", "T", &fused_conv);
[all …]
/external/tensorflow/tensorflow/python/grappler/
Dcost_analyzer_tool.py26 from tensorflow.contrib.fused_conv.ops import gen_fused_conv2d_bias_activation_op # pylint: disabl…
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_test.cc222 Output fused_conv = FusedResizeAndPadConv2D( in CompareFusedAndSeparate() local
276 Output fused_conv = FusedPadConv2D( in CompareFusedPadOnlyAndSeparate() local