Home
last modified time | relevance | path

Searched refs:conv2d (Results 1 – 25 of 103) sorted by relevance

12345

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_space_to_depth_pass.cc150 void HandleConv2DStride(TF::Conv2DOp conv2d) { in HandleConv2DStride() argument
151 MLIRContext* context = conv2d.getContext(); in HandleConv2DStride()
158 conv2d->setAttr("strides", strides); in HandleConv2DStride()
162 void HandleConv2DInput(TF::Conv2DOp conv2d, int64_t block_size) { in HandleConv2DInput() argument
163 auto input = conv2d.input(); in HandleConv2DInput()
217 void HandleConv2DFilter(TF::Conv2DOp conv2d, int64_t block_size) { in HandleConv2DFilter() argument
224 auto filter = conv2d.filter(); in HandleConv2DFilter()
225 OpBuilder builder(conv2d); in HandleConv2DFilter()
226 builder.setInsertionPoint(conv2d); in HandleConv2DFilter()
264 conv2d.setOperand(1, final_reshape_op); in HandleConv2DFilter()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_benchmark_test.cc46 Node* conv2d; member
51 Node* conv2d; member
57 Node* conv2d; member
64 Node* conv2d; member
70 Node* conv2d; member
100 Node* conv2d; in Conv2D() local
111 .Finalize(graph, &conv2d)); in Conv2D()
113 return {graph, conv2d}; in Conv2D()
125 Node* conv2d = conv_graph.conv2d; in Conv2DWithBias() local
132 .Input(conv2d) in Conv2DWithBias()
[all …]
Dconv_ops_fused_impl.h251 LaunchFusedConv2DWithOutputKernel<T> conv2d(
264 conv2d(WithBiasAdd<T>(bias_add_args), context, input, filter, output);
267 conv2d(WithBiasAddAndRelu<T>(bias_add_args), context, input, filter,
271 conv2d(WithBiasAddAndRelu6<T>(bias_add_args), context, input, filter,
275 conv2d(WithBiasAddAndLeakyRelu<T>(bias_add_args), context, input,
279 conv2d(WithBiasAddAndElu<T>(bias_add_args), context, input, filter,
283 conv2d(
288 conv2d(WithFusedBatchNormAndRelu<T>(fusion_args.epsilon,
293 conv2d(WithFusedBatchNormAndRelu6<T>(fusion_args.epsilon,
298 conv2d(WithFusedBatchNormAndLeakyRelu<T>(fusion_args.epsilon,
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc111 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 …]
/external/tensorflow/tensorflow/python/layers/
Dconvolutional.py23 conv2d = convolutional.conv2d variable
44 convolution2d = conv2d
/external/tensorflow/tensorflow/python/profiler/internal/
Dmodel_analyzer_testlib.py45 x = nn_ops.conv2d(image, kernel, [1, 2, 2, 1], padding='SAME')
50 x = nn_ops.conv2d(x, kernel, [1, 2, 2, 1], padding='SAME')
79 r1 = nn_ops.conv2d(image, kernel1, [1, 2, 2, 1], padding='SAME')
85 r2 = nn_ops.conv2d(image, kernel2, [1, 2, 2, 1], padding='SAME')
Dprint_model_analysis_test.py56 x = nn_ops.conv2d(image, kernel, [1, 2, 2, 1], padding='SAME')
/external/tensorflow/tensorflow/python/tools/
Doptimize_for_inference_test.py136 conv_op = nn_ops.conv2d(
174 ("NHWC", False, nn_ops.conv2d), ("NCHW", True, nn_ops.conv2d),
184 if conv2d_func == nn_ops.conv2d:
234 for data_format, conv2d_func in [("NHWC", nn_ops.conv2d),
235 ("NCHW", nn_ops.conv2d),
244 if conv2d_func == nn_ops.conv2d:
305 nn_ops.conv2d(
335 nn_ops.conv2d(
365 nn_ops.conv2d(
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dconst_broadcast_test.py32 y1 = nn.conv2d(x, filt1, strides=[1, 1, 1, 1], padding='SAME', name='y1')
36 y2 = nn.conv2d(z1, filt2, strides=[1, 1, 1, 1], padding='SAME', name='y2')
40 y3 = nn.conv2d(z2, filt3, strides=[1, 1, 1, 1], padding='SAME', name='y3')
Dmemory_alignment_test.py38 conv = nn.conv2d(
44 out = nn.conv2d(
Ddynamic_input_shapes_test.py34 x = nn.conv2d(
45 x = nn.conv2d(
/external/tensorflow/tensorflow/python/ops/
Dconv2d_benchmark.py68 conv2d_op = nn_ops.conv2d(
73 conv2d_op = nn_ops.conv2d(
78 warmup_conv2d_op = nn_ops.conv2d(
83 warmup_conv2d_op = nn_ops.conv2d(
/external/tensorflow/tensorflow/lite/testing/op_tests/
Dconv_bias_activation.py66 out = tf.nn.conv2d(
93 out = tf.nn.conv2d(
111 out = tf.nn.conv2d(
Dconv_to_depthwiseconv_with_shared_weights.py58 result1 = tf.nn.conv2d(
65 result2 = tf.nn.conv2d(
Dconv_with_shared_weights.py63 result1 = tf.nn.conv2d(
70 result2 = tf.nn.conv2d(
/external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/
Dconv_bias_activation.py70 out = tf.nn.conv2d(
97 out = tf.nn.conv2d(
115 out = tf.nn.conv2d(
/external/armnn/src/armnn/test/optimizations/
DConvertConstDequantisationLayersToConstLayersTest.cpp111 auto conv2d = graph.AddLayer<Convolution2dLayer>(convolution2dDescriptor, "conv2d"); variable
132 dequantizeLayer->GetOutputSlot().Connect(conv2d->GetInputSlot(1));
134 padLayer->GetOutputSlot().Connect(conv2d->GetInputSlot(0));
135 conv2d->GetOutputSlot().Connect(output->GetInputSlot(0));
/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
Dconv_ops_test.py216 conv = nn_ops.conv2d(
252 conv = nn_ops.conv2d(
291 computed = nn_ops.conv2d(
404 conv2d_result = nn_ops.conv2d(
442 conv1 = nn_ops.conv2d(
447 conv2 = nn_ops.conv2d(
826 return nn_ops.conv2d(
1333 conv_forward = nn_ops.conv2d(
1379 conv_forward = nn_ops.conv2d(
1939 conv = nn_ops.conv2d(
[all …]
Datrous_conv2d_test.py78 y2 = nn_ops.conv2d(
129 y2 = nn_ops.conv2d(y2, f, strides=[1, 1, 1, 1], padding=padding)
130 y2 = nn_ops.conv2d(y2, f, strides=[1, 1, 1, 1], padding=padding)
131 y2 = nn_ops.conv2d(y2, f, strides=[1, 1, 1, 1], padding=padding)
Dconv2d_backprop_filter_grad_test.py50 conv_out = nn_ops.conv2d(
93 conv_out = nn_ops.conv2d(
/external/armnn/src/profiling/test/
DProfilingTestUtils.cpp428 IConnectableLayer* conv2d = net->AddConvolution2dLayer(conv2dDesc); in VerifyPostOptimisationStructureTestImpl() local
434 weightsLayer->GetOutputSlot(0).Connect(conv2d->GetInputSlot(1u)); in VerifyPostOptimisationStructureTestImpl()
437 biasLayer->GetOutputSlot(0).Connect(conv2d->GetInputSlot(2u)); in VerifyPostOptimisationStructureTestImpl()
446 input->GetOutputSlot(0).Connect(conv2d->GetInputSlot(0)); in VerifyPostOptimisationStructureTestImpl()
447 conv2d->GetOutputSlot(0).Connect(abs->GetInputSlot(0)); in VerifyPostOptimisationStructureTestImpl()
451 conv2d->GetOutputSlot(0).SetTensorInfo(outputInfo); in VerifyPostOptimisationStructureTestImpl()
714 VerifyTimelineEntityBinaryPacketData(conv2d->GetGuid(), readableData, offset); in VerifyPostOptimisationStructureTestImpl()
723 conv2d->GetGuid(), in VerifyPostOptimisationStructureTestImpl()
732 conv2d->GetGuid(), in VerifyPostOptimisationStructureTestImpl()
742 conv2d->GetGuid(), in VerifyPostOptimisationStructureTestImpl()
[all …]
/external/tensorflow/tensorflow/examples/speech_commands/
Dmodels.py269 first_conv = tf.nn.conv2d(input=fingerprint_4d,
296 second_conv = tf.nn.conv2d(input=max_pool,
394 first_conv = tf.nn.conv2d(
727 first_conv = tf.nn.conv2d(
834 first_conv = tf.nn.conv2d(
861 second_conv = tf.nn.conv2d(
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops_test.cc111 Node* conv2d; in DefaultConv2D() local
118 .Finalize(graph, &conv2d)); in DefaultConv2D()
135 Node* conv2d; in MklConv2D() local
145 .Finalize(graph, &conv2d)); in MklConv2D()
/external/XNNPACK/src/operators/
Dconvolution-nchw.c435 convolution_op->ukernel.conv2d = (struct xnn_ukernel_conv2d) { in xnn_create_convolution2d_nchw_f32()
661 convolution_op->context.conv2d = (struct conv2d_context) { in setup_convolution2d_nchw()
674 .hwc2chw_ukernel = convolution_op->ukernel.conv2d.hwc2chw_function, in setup_convolution2d_nchw()
676 …memcpy(&convolution_op->context.conv2d.params, params, sizeof(convolution_op->context.conv2d.param… in setup_convolution2d_nchw()
678 const size_t output_height_tile = convolution_op->ukernel.conv2d.output_height_tile; in setup_convolution2d_nchw()
/external/tensorflow/tensorflow/python/grappler/
Dremapper_test.py231 y = nn_ops.conv2d(
259 y = nn_ops.conv2d(
263 y = nn_ops.conv2d(

12345