| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_space_to_depth_pass.cc | 195 void HandleConv2DStride(TF::Conv2DOp conv2d) { in HandleConv2DStride() argument
196 MLIRContext* context = conv2d.getContext(); in HandleConv2DStride()
203 conv2d->setAttr("strides", strides); in HandleConv2DStride()
207 void HandleConv2DInput(TF::Conv2DOp conv2d, int64_t block_size) { in HandleConv2DInput() argument
208 auto input = conv2d.input(); in HandleConv2DInput()
262 void HandleConv2DFilter(TF::Conv2DOp conv2d, int64_t block_size) { in HandleConv2DFilter() argument
269 auto filter = conv2d.filter(); in HandleConv2DFilter()
270 OpBuilder builder(conv2d); in HandleConv2DFilter()
271 builder.setInsertionPoint(conv2d); in HandleConv2DFilter()
309 conv2d.setOperand(1, final_reshape_op); in HandleConv2DFilter()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_benchmark_test.cc | 38 Node* conv2d; member
43 Node* conv2d; member
49 Node* conv2d; member
56 Node* conv2d; member
62 Node* conv2d; member
92 Node* conv2d; in Conv2D() local
100 .Finalize(graph, &conv2d)); in Conv2D()
102 return {graph, conv2d}; in Conv2D()
114 Node* conv2d = conv_graph.conv2d; in Conv2DWithBias() local
121 .Input(conv2d) in Conv2DWithBias()
[all …]
|
| D | conv_ops_fused_impl.h | 246 LaunchFusedConv2DWithOutputKernel<T> conv2d(
256 conv2d(WithBiasAdd<T>(bias_add_args), context, input, filter, output);
259 conv2d(WithBiasAddAndRelu<T>(bias_add_args), context, input, filter,
263 conv2d(WithBiasAddAndRelu6<T>(bias_add_args), context, input, filter,
267 conv2d(WithBiasAddAndLeakyRelu<T>(bias_add_args), context, input,
271 conv2d(WithBiasAddAndElu<T>(bias_add_args), context, input, filter,
275 conv2d(
280 conv2d(WithFusedBatchNormAndRelu<T>(fusion_args.epsilon,
285 conv2d(WithFusedBatchNormAndRelu6<T>(fusion_args.epsilon,
290 conv2d(WithFusedBatchNormAndLeakyRelu<T>(fusion_args.epsilon,
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 111 auto conv2d = ops::Conv2D( in SimpleConv2D() local
115 return conv2d; in SimpleConv2D()
121 auto conv2d = SimpleConv2D(&scope, data_type); in CreateSimpleConv2DGraph() local
122 auto output = ops::Identity(scope.WithOpName("output"), conv2d); in CreateSimpleConv2DGraph()
296 Output conv2d = ops::Conv2D( in CreateSimpleAddN() local
306 {a, b, c, conv2d}); in CreateSimpleAddN()
387 Output conv2d = ops::Conv2D( in ReduceTransposerKeepDims() local
394 conv2d, axis, attrs); in ReduceTransposerKeepDims()
455 Output conv2d = ops::Conv2D( in ReduceTransposerValidAxisNode() local
461 conv2d, axis); in ReduceTransposerValidAxisNode()
[all …]
|
| D | remapper.cc | 263 bool IsCpuCompatibleDataFormat(const NodeDef* conv2d) { in IsCpuCompatibleDataType() argument
264 DCHECK(IsConv2D(*conv2d)) << "Expected Conv2D op"; in IsCpuCompatibleDataType()
265 const string& data_format = conv2d->attr().at(kDataFormat).s(); in IsCpuCompatibleDataType()
273 bool IsGpuCompatibleDataFormat(const NodeDef* conv2d) { in IsCpuCompatibleDataType() argument
274 DCHECK(IsConv2D(*conv2d)) << "Expected Conv2D op"; in IsCpuCompatibleDataType()
275 const string& data_format = conv2d->attr().at(kDataFormat).s(); in IsCpuCompatibleDataType()
279 bool IsCpuCompatibleConv2D(const NodeDef* conv2d) { in IsCpuCompatibleDataType() argument
280 DCHECK(IsConv2D(*conv2d)) << "Expected Conv2D op"; in IsCpuCompatibleDataType()
281 return NodeIsOnCpu(conv2d) && IsCpuCompatibleDataType(conv2d) && in IsCpuCompatibleDataType()
282 IsCpuCompatibleDataFormat(conv2d); in IsCpuCompatibleDataType()
[all …]
|
| /external/tensorflow/tensorflow/python/layers/ |
| D | convolutional.py | 27 conv2d = convolutional.conv2d variable
48 convolution2d = conv2d
|
| /external/tensorflow/tensorflow/python/profiler/internal/ |
| D | model_analyzer_testlib.py | 49 x = nn_ops.conv2d(image, kernel, [1, 2, 2, 1], padding='SAME')
54 x = nn_ops.conv2d(x, kernel, [1, 2, 2, 1], padding='SAME')
83 r1 = nn_ops.conv2d(image, kernel1, [1, 2, 2, 1], padding='SAME')
89 r2 = nn_ops.conv2d(image, kernel2, [1, 2, 2, 1], padding='SAME')
|
| D | print_model_analysis_test.py | 60 x = nn_ops.conv2d(image, kernel, [1, 2, 2, 1], padding='SAME')
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | const_broadcast_test.py | 36 y1 = nn.conv2d(x, filt1, strides=[1, 1, 1, 1], padding='SAME', name='y1')
40 y2 = nn.conv2d(z1, filt2, strides=[1, 1, 1, 1], padding='SAME', name='y2')
44 y3 = nn.conv2d(z2, filt3, strides=[1, 1, 1, 1], padding='SAME', name='y3')
|
| D | memory_alignment_test.py | 40 conv = nn.conv2d(
46 out = nn.conv2d(
|
| D | dynamic_input_shapes_test.py | 38 x = nn.conv2d(
49 x = nn.conv2d(
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | optimize_for_inference_test.py | 140 conv_op = nn_ops.conv2d(
178 ("NHWC", False, nn_ops.conv2d), ("NCHW", True, nn_ops.conv2d),
188 if conv2d_func == nn_ops.conv2d:
238 for data_format, conv2d_func in [("NHWC", nn_ops.conv2d),
239 ("NCHW", nn_ops.conv2d),
248 if conv2d_func == nn_ops.conv2d:
309 nn_ops.conv2d(
339 nn_ops.conv2d(
369 nn_ops.conv2d(
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | conv2d_benchmark.py | 72 conv2d_op = nn_ops.conv2d(
77 conv2d_op = nn_ops.conv2d(
82 warmup_conv2d_op = nn_ops.conv2d(
87 warmup_conv2d_op = nn_ops.conv2d(
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | conv_ops_test.py | 217 conv = nn_ops.conv2d(
253 conv = nn_ops.conv2d(
292 computed = nn_ops.conv2d(
405 conv2d_result = nn_ops.conv2d(
443 conv1 = nn_ops.conv2d(
448 conv2 = nn_ops.conv2d(
818 return nn_ops.conv2d(
1304 conv_forward = nn_ops.conv2d(
1350 conv_forward = nn_ops.conv2d(
1910 conv = nn_ops.conv2d(
[all …]
|
| D | atrous_conv2d_test.py | 82 y2 = nn_ops.conv2d(
133 y2 = nn_ops.conv2d(y2, f, strides=[1, 1, 1, 1], padding=padding)
134 y2 = nn_ops.conv2d(y2, f, strides=[1, 1, 1, 1], padding=padding)
135 y2 = nn_ops.conv2d(y2, f, strides=[1, 1, 1, 1], padding=padding)
|
| D | conv2d_backprop_filter_grad_test.py | 52 conv_out = nn_ops.conv2d(
95 conv_out = nn_ops.conv2d(
|
| /external/llvm-project/mlir/test/Dialect/Tosa/ |
| D | quant-test.mlir | 14 // CHECK: tosa.conv2d
15 …%0 = "tosa.conv2d"(%arg0, %arg1, %arg2) {pad = [1, 1, 2, 2], dilation = [2, 1], stride = [1, 1], q…
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | conv_to_depthwiseconv_with_shared_weights.py | 62 result1 = tf.nn.conv2d(
69 result2 = tf.nn.conv2d(
|
| D | conv_with_shared_weights.py | 67 result1 = tf.nn.conv2d(
74 result2 = tf.nn.conv2d(
|
| /external/tensorflow/tensorflow/python/keras/legacy_tf_layers/ |
| D | convolutional_test.py | 43 conv_layers.conv2d(images, 32, 3, data_format='invalid')
49 conv_layers.conv2d(images, 32, 3, strides=(1, 2, 3))
52 conv_layers.conv2d(images, 32, 3, strides=None)
58 conv_layers.conv2d(images, 32, (1, 2, 3))
61 conv_layers.conv2d(images, 32, None)
78 output = conv_layers.conv2d(images, 32, [3, 3], activation=nn_ops.relu)
268 conv_layers.conv2d(images, 32, [3, 3], name='conv1')
270 conv_layers.conv2d(images, 32, [3, 3], name='conv1', reuse=True)
278 conv_layers.conv2d(images, 32, [3, 3], name='conv1')
281 conv_layers.conv2d(images, 32, [3, 3], name='conv1')
[all …]
|
| /external/tensorflow/tensorflow/examples/speech_commands/ |
| D | models.py | 273 first_conv = tf.nn.conv2d(input=fingerprint_4d,
300 second_conv = tf.nn.conv2d(input=max_pool,
398 first_conv = tf.nn.conv2d(
731 first_conv = tf.nn.conv2d(
838 first_conv = tf.nn.conv2d(
865 second_conv = tf.nn.conv2d(
|
| /external/XNNPACK/src/operators/ |
| D | convolution-nchw.c | 433 convolution_op->ukernel.conv2d = (struct xnn_ukernel_conv2d) { in xnn_create_convolution2d_nchw_f32()
641 convolution_op->context.conv2d = (struct conv2d_context) { in setup_convolution2d_nchw()
654 .hwc2chw_ukernel = convolution_op->ukernel.conv2d.hwc2chw_function, in setup_convolution2d_nchw()
656 …memcpy(&convolution_op->context.conv2d.params, params, sizeof(convolution_op->context.conv2d.param… in setup_convolution2d_nchw()
659 const size_t output_height_tile = convolution_op->ukernel.conv2d.output_height_tile; in setup_convolution2d_nchw()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops_test.cc | 114 Node* conv2d; in DefaultConv2D() local
121 .Finalize(graph, &conv2d)); in DefaultConv2D()
139 Node* conv2d; in MklConv2D() local
149 .Finalize(graph, &conv2d)); in MklConv2D()
|
| /external/XNNPACK/src/xnnpack/ |
| D | operator.h | 161 struct xnn_ukernel_conv2d conv2d; member
311 struct conv2d_context conv2d; member
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/tests/ |
| D | fuse-bias-tf.mlir | 10 // CHECK: tosa.conv2d
|