Searched refs:fused_conv (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fuse_convolutions.cc | 61 NodeDef fused_conv; in FuseResizePadAndConv() local
62 fused_conv.set_op("FusedResizeAndPadConv2D"); in FuseResizePadAndConv()
63 fused_conv.set_name(match.node.name()); in FuseResizePadAndConv()
64 AddNodeInput(resize_node.input(0), &fused_conv); in FuseResizePadAndConv()
65 AddNodeInput(resize_node.input(1), &fused_conv); in FuseResizePadAndConv()
66 AddNodeInput(mirror_pad_node.input(1), &fused_conv); in FuseResizePadAndConv()
67 AddNodeInput(conv_node.input(1), &fused_conv); in FuseResizePadAndConv()
69 &fused_conv); in FuseResizePadAndConv()
70 CopyNodeAttr(mirror_pad_node, "mode", "mode", &fused_conv); in FuseResizePadAndConv()
71 CopyNodeAttr(conv_node, "T", "T", &fused_conv); in FuseResizePadAndConv()
[all …]
|
| D | fuse_quantized_convolution.cc | 73 NodeDef fused_conv; in FuseQuantizedConvolutionAndRequantize() local
74 fused_conv.set_op(quantized_conv2D_op_name + "AndRequantize"); in FuseQuantizedConvolutionAndRequantize()
75 fused_conv.set_name(match.node.name()); in FuseQuantizedConvolutionAndRequantize()
82 AddNodeInput(quantized_conv2D_node.input(i), &fused_conv); in FuseQuantizedConvolutionAndRequantize()
84 AddNodeInput(const_requantize_range_min_node.name(), &fused_conv); in FuseQuantizedConvolutionAndRequantize()
85 AddNodeInput(const_requantize_range_max_node.name(), &fused_conv); in FuseQuantizedConvolutionAndRequantize()
96 AddNodeInput(summand, &fused_conv); in FuseQuantizedConvolutionAndRequantize()
97 AddNodeInput(min_summand, &fused_conv); in FuseQuantizedConvolutionAndRequantize()
98 AddNodeInput(max_summand, &fused_conv); in FuseQuantizedConvolutionAndRequantize()
110 fused_conv.set_op( in FuseQuantizedConvolutionAndRequantize()
[all …]
|
| D | README.md | 990 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/contrib/fused_conv/ |
| D | __init__.py | 22 from tensorflow.contrib.fused_conv.python.ops.fused_conv2d_bias_activation_op import *
|
| /external/tensorflow/tensorflow/contrib/fused_conv/python/ops/ |
| D | fused_conv2d_bias_activation_op_test.py | 22 from tensorflow.contrib.fused_conv.python.ops import fused_conv2d_bias_activation_op_test_base as t…
|
| D | fused_conv2d_bias_activation_op.py | 21 from tensorflow.contrib.fused_conv.ops import gen_fused_conv2d_bias_activation_op
|
| D | fused_conv2d_bias_activation_benchmark.py | 22 from tensorflow.contrib.fused_conv.python.ops import fused_conv2d_bias_activation_op
|
| D | fused_conv2d_bias_activation_op_test_base.py | 33 from tensorflow.contrib.fused_conv.python.ops import fused_conv2d_bias_activation_op
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cost_analyzer_tool.py | 26 from tensorflow.contrib.fused_conv.ops import gen_fused_conv2d_bias_activation_op # pylint: disabl…
|
| /external/tensorflow/tensorflow/contrib/cmake/ |
| D | python_modules.txt | 176 tensorflow/contrib/fused_conv
177 tensorflow/contrib/fused_conv/kernels
178 tensorflow/contrib/fused_conv/python
179 tensorflow/contrib/fused_conv/python/ops
|
| /external/tensorflow/tensorflow/contrib/ |
| D | BUILD | 138 "//tensorflow/contrib/fused_conv:fused_conv_py",
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_test.cc | 219 Output fused_conv = FusedResizeAndPadConv2D( in CompareFusedAndSeparate() local
273 Output fused_conv = FusedPadConv2D( in CompareFusedPadOnlyAndSeparate() local
|