Home
last modified time | relevance | path

Searched defs:conv2d (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc111 auto conv2d = ops::Conv2D( in SimpleConv2D() local
121 auto conv2d = SimpleConv2D(&scope, data_type); in CreateSimpleConv2DGraph() local
296 Output conv2d = ops::Conv2D( in CreateSimpleAddN() local
387 Output conv2d = ops::Conv2D( in ReduceTransposerKeepDims() local
455 Output conv2d = ops::Conv2D( in ReduceTransposerValidAxisNode() local
602 auto* conv2d = context.graph_view->GetNode("conv2d"); in TEST_F() local
636 auto* conv2d = context.graph_view->GetNode("conv2d"); in TEST_F() local
735 auto* conv2d = context.graph_view->GetNode("conv2d"); in TEST_F() local
840 auto* conv2d = context.graph_view->GetNode("conv2d"); in TEST_F() local
1297 auto conv2d = SimpleConv2D(&scope); in TEST_F() local
[all …]
Dremapper.cc343 const NodeDef* conv2d) { in IsGpuCompatibleDataFormat()
349 bool IsCpuCompatibleConv2D(const RemapperContext& ctx, const NodeDef* conv2d) { in IsCpuCompatibleConv2D()
361 bool IsGpuCompatibleConv2D(const RemapperContext& ctx, const NodeDef* conv2d) { in IsGpuCompatibleConv2D()
2062 void CopyConv2DAttributes(const NodeDef& conv2d, NodeDef* fused_conv2d, in CopyConv2DAttributes()
Dgeneric_layout_optimizer_test.cc233 auto conv2d = SimpleConv2D(&scope, 4, 2, "VALID", ""); in TEST_F() local
/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
125 Node* conv2d = conv_graph.conv2d; in Conv2DWithBias() local
153 Node* conv2d = conv_graph.conv2d; in Conv2DWithBiasAndActivation() local
174 Node* conv2d = conv_graph.conv2d; in Conv2DWithBatchNorm() local
213 Node* conv2d = conv_graph.conv2d; in Conv2DWithBatchNormAndActivation() local
Dconv_grad_input_ops_benchmark_test.cc79 Node* conv2d; in Conv2DBackpropInput() local
Dconv_grad_filter_ops_benchmark_test.cc80 Node* conv2d; in Conv2DBackpropFilter() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_space_to_depth_pass.cc150 void HandleConv2DStride(TF::Conv2DOp conv2d) { in HandleConv2DStride()
162 void HandleConv2DInput(TF::Conv2DOp conv2d, int64_t block_size) { in HandleConv2DInput()
217 void HandleConv2DFilter(TF::Conv2DOp conv2d, int64_t block_size) { in HandleConv2DFilter()
527 Optional<BlockArgumentInfo> GetConv2DInputArgNum(TF::Conv2DOp conv2d) { in GetConv2DInputArgNum()
537 void HandleFirstConvolution(TF::Conv2DOp conv2d, int64_t block_size) { in HandleFirstConvolution()
589 int32_t GetConv2DBlockSize(TF::Conv2DOp conv2d) { in GetConv2DBlockSize()
629 auto conv2d_result = device_func.walk([&](TF::Conv2DOp conv2d) { in runOnOperation()
/external/tensorflow/tensorflow/python/layers/
Dconvolutional.py23 conv2d = convolutional.conv2d variable
/external/armnn/src/armnn/test/optimizations/
DConvertConstDequantisationLayersToConstLayersTest.cpp111 auto conv2d = graph.AddLayer<Convolution2dLayer>(convolution2dDescriptor, "conv2d"); variable
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops_test.cc111 Node* conv2d; in DefaultConv2D() local
135 Node* conv2d; in MklConv2D() local
/external/XNNPACK/src/xnnpack/
Doperator.h90 struct xnn_ukernel_conv2d conv2d; member
274 struct conv2d_context conv2d; member
/external/armnn/src/profiling/test/
DProfilingTestUtils.cpp428 IConnectableLayer* conv2d = net->AddConvolution2dLayer(conv2dDesc); in VerifyPostOptimisationStructureTestImpl() local
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dconvolutional.py328 def conv2d(inputs, function
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py2363 def conv2d( # pylint: disable=redefined-builtin,dangerous-default-value function
/external/tensorflow/tensorflow/python/keras/
Dbackend.py5306 def conv2d(x, function
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc866 string conv2d; member
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc6050 auto conv2d = ops::Conv2D(s.WithOpName("my_conv2d"), input, filter, strides, in TEST_P() local
6271 auto conv2d = ops::Conv2DBackpropInput( in TEST_P() local