Searched refs:convolution2d (Results 1 – 12 of 12) sorted by relevance
| /external/pytorch/aten/src/ATen/native/xnnpack/ |
| D | RegisterOpContextClass.cpp | 12 using internal::convolution2d::createConv2dClampPrePackOpContext;
13 using internal::convolution2d::createConv2dTransposeClampPrePackOpContext;
76 …d(Any W_prepack) -> (Any)"), [](const IValue& inp) { return internal::convolution2d::unpack_prepac… in TORCH_LIBRARY()
91 …m.impl(TORCH_SELECTIVE_NAME("prepacked::conv2d_clamp_run"), TORCH_FN(internal::convolution2d::conv… in TORCH_LIBRARY_IMPL()
92 …IVE_NAME("prepacked::conv2d_transpose_clamp_run"), TORCH_FN(internal::convolution2d::conv2d_transp… in TORCH_LIBRARY_IMPL()
|
| D | OpContext.cpp | 57 xnnpack::internal::convolution2d::create( in create_context()
101 xnnpack::internal::convolution2d::create( in create_context()
137 return xnnpack::internal::convolution2d::run(op_context_, input); in run()
142 return xnnpack::internal::convolution2d::run(op_context_, input); in run()
|
| D | Convolution.h | 10 namespace internal::convolution2d {
62 Tensor convolution2d(
|
| D | Convolution.cpp | 15 namespace convolution2d { namespace
479 return internal::convolution2d::available( in use_convolution2d()
489 internal::convolution2d::usable(input); in use_convolution2d()
492 Tensor convolution2d( in convolution2d() function
500 return internal::convolution2d::create_and_run( in convolution2d()
|
| D | Shim.cpp | 43 Tensor convolution2d( in convolution2d() function
|
| D | Engine.h | 22 Tensor convolution2d(
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | Convolution2dEndToEndTestImpl.hpp | 35 IConnectableLayer* convolution2d = network->AddConvolution2dLayer(descriptor, "convolution2d"); in CreateConstConvolution2dNetwork() local
38 Connect(input, convolution2d, inputInfo, 0, 0); in CreateConstConvolution2dNetwork()
39 Connect(weightsLayer, convolution2d, weightsInfo, 0, 1); in CreateConstConvolution2dNetwork()
44 Connect(biasLayer, convolution2d, biasInfo, 0, 2); in CreateConstConvolution2dNetwork()
47 Connect(convolution2d, output, outputInfo, 0, 0); in CreateConstConvolution2dNetwork()
|
| D | DepthwiseConvolution2dEndToEndTests.hpp | 35 …IConnectableLayer* convolution2d = network->AddDepthwiseConvolution2dLayer(descriptor, "depthwiseC… in CreateDepthwiseConvolution2dNetwork() local
38 Connect(input, convolution2d, inputInfo, 0, 0); in CreateDepthwiseConvolution2dNetwork()
39 Connect(weightsLayer, convolution2d, weightsInfo, 0, 1); in CreateDepthwiseConvolution2dNetwork()
40 Connect(biasLayer, convolution2d, biasInfo, 0, 2); in CreateDepthwiseConvolution2dNetwork()
41 Connect(convolution2d, output, outputInfo, 0, 0); in CreateDepthwiseConvolution2dNetwork()
|
| /external/tensorflow/tensorflow/python/layers/ |
| D | convolutional.py | 44 convolution2d = conv2d variable
|
| /external/pytorch/aten/src/ATen/test/ |
| D | xnnpack_test.cpp | 156 const auto result = at::native::xnnpack::convolution2d( in TEST()
|
| /external/tensorflow/tensorflow/python/keras/legacy_tf_layers/ |
| D | convolutional.py | 1518 convolution2d = conv2d variable
|
| /external/pytorch/aten/src/ATen/native/ |
| D | Convolution.cpp | 1639 output = xnnpack::convolution2d( in _convolution()
|