Home
last modified time | relevance | path

Searched defs:conv2d (Results 1 – 15 of 15) 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.cc263 bool IsCpuCompatibleDataFormat(const NodeDef* conv2d) { in IsCpuCompatibleDataType()
273 bool IsGpuCompatibleDataFormat(const NodeDef* conv2d) { in IsCpuCompatibleDataType()
279 bool IsCpuCompatibleConv2D(const NodeDef* conv2d) { in IsCpuCompatibleDataType()
285 bool IsGpuCompatibleConv2D(const NodeDef* conv2d) { in IsCpuCompatibleDataType()
965 void CopyConv2DAttributes(const NodeDef& conv2d, NodeDef* fused_conv2d, in IsCpuCompatibleDataType()
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.cc38 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
114 Node* conv2d = conv_graph.conv2d; in Conv2DWithBias() local
142 Node* conv2d = conv_graph.conv2d; in Conv2DWithBiasAndActivation() local
163 Node* conv2d = conv_graph.conv2d; in Conv2DWithBatchNorm() local
202 Node* conv2d = conv_graph.conv2d; in Conv2DWithBatchNormAndActivation() local
Dconv_grad_filter_ops_benchmark_test.cc80 Node* conv2d; in Conv2DBackpropFilter() local
Dconv_grad_input_ops_benchmark_test.cc79 Node* conv2d; in Conv2DBackpropInput() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_space_to_depth_pass.cc195 void HandleConv2DStride(TF::Conv2DOp conv2d) { in HandleConv2DStride()
207 void HandleConv2DInput(TF::Conv2DOp conv2d, int64_t block_size) { in HandleConv2DInput()
262 void HandleConv2DFilter(TF::Conv2DOp conv2d, int64_t block_size) { in HandleConv2DFilter()
572 Optional<BlockArgumentInfo> GetConv2DInputArgNum(TF::Conv2DOp conv2d) { in GetConv2DInputArgNum()
582 void HandleFirstConvolution(TF::Conv2DOp conv2d, int64_t block_size) { in HandleFirstConvolution()
634 int32_t GetConv2DBlockSize(TF::Conv2DOp conv2d) { in GetConv2DBlockSize()
674 auto conv2d_result = device_func.walk([&](TF::Conv2DOp conv2d) { in runOnOperation()
/external/tensorflow/tensorflow/python/layers/
Dconvolutional.py27 conv2d = convolutional.conv2d variable
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops_test.cc114 Node* conv2d; in DefaultConv2D() local
139 Node* conv2d; in MklConv2D() local
/external/XNNPACK/src/xnnpack/
Doperator.h161 struct xnn_ukernel_conv2d conv2d; member
311 struct conv2d_context conv2d; member
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dconvolutional.py328 def conv2d(inputs, function
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py2174 def conv2d( # pylint: disable=redefined-builtin,dangerous-default-value function
/external/tensorflow/tensorflow/python/keras/
Dbackend.py5379 def conv2d(x, function
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc916 string conv2d; member
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc4355 auto conv2d = ops::Conv2D(s.WithOpName("my_conv2d"), input, filter, strides, in TEST_P() local
4578 auto conv2d = ops::Conv2DBackpropInput( in TEST_F() local