| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | generic_layout_optimizer_transposer_test.cc | 111 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 …]
|
| H A D | remapper.cc | 343 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()
|
| H A D | generic_layout_optimizer_test.cc | 233 auto conv2d = SimpleConv2D(&scope, 4, 2, "VALID", ""); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_ops_benchmark_test.cc | 46 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
|
| H A D | conv_grad_filter_ops_benchmark_test.cc | 80 Node* conv2d; in Conv2DBackpropFilter() local
|
| H A D | conv_grad_input_ops_benchmark_test.cc | 79 Node* conv2d; in Conv2DBackpropInput() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | tpu_space_to_depth_pass.cc | 150 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_conv_ops_test.cc | 111 Node* conv2d; in DefaultConv2D() local
135 Node* conv2d; in MklConv2D() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/layers/ |
| H A D | convolutional.py | 23 conv2d = convolutional.conv2d variable
|
| /aosp_15_r20/external/armnn/src/armnn/test/optimizations/ |
| H A D | ConvertConstDequantisationLayersToConstLayersTest.cpp | 111 auto conv2d = graph.AddLayer<Convolution2dLayer>(convolution2dDescriptor, "conv2d"); variable
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | conv.h | 72 inline Tensor conv2d( in conv2d() function
|
| /aosp_15_r20/external/XNNPACK/src/xnnpack/ |
| H A D | operator.h | 90 struct xnn_ukernel_conv2d conv2d; member
274 struct conv2d_context conv2d; member
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | android_api_module.py | 107 def conv2d(self, x: Tensor, w: Tensor, toChannelsLast: bool) -> Tensor: member in AndroidAPIModule
|
| /aosp_15_r20/external/pytorch/android/pytorch_android/ |
| H A D | generate_test_torchscripts.py | 121 def conv2d(self, x: Tensor, w: Tensor, toChannelsLast: bool) -> Tensor: member in Test
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | graph_rewrite_helper.cpp | 79 std::string conv2d = R"( in replaceConvolutionWithAtenConv() local
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_caffe2.py | 93 def conv2d( function
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/ |
| H A D | functional.py | 245 def conv2d( function
|
| /aosp_15_r20/external/armnn/src/profiling/test/ |
| H A D | ProfilingTestUtils.cpp | 428 IConnectableLayer* conv2d = net->AddConvolution2dLayer(conv2dDesc); in VerifyPostOptimisationStructureTestImpl() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | quantization_patterns.h | 331 std::string conv2d = R"( in quant_fusion_pattern_and_replacements() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/ |
| H A D | convolutional.py | 328 def conv2d(inputs, function
|
| /aosp_15_r20/external/pytorch/torch/jit/ |
| H A D | _shape_functions.py | 763 def conv2d( function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Convolution.cpp | 29 namespace conv2d { namespace
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | nn_ops.py | 2363 def conv2d( # pylint: disable=redefined-builtin,dangerous-default-value function
|
| /aosp_15_r20/external/pytorch/torch/nn/ |
| H A D | functional.py | 92 conv2d = _add_docstr( variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/ |
| H A D | backend.py | 5306 def conv2d(x, function
|