Home
last modified time | relevance | path

Searched defs:avg_pool2d (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/layers/
H A Dpooling.py39 avg_pool2d = average_pooling2d variable
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DAveragePool2d.cpp17 TORCH_PRECOMPUTE_META_FUNC(avg_pool2d) in TORCH_PRECOMPUTE_META_FUNC() argument
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dpooling.h53 inline Tensor avg_pool2d( in avg_pool2d() function
85 inline Tensor avg_pool2d( in avg_pool2d() function
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DPool.cpp192 void avg_pool2d(ComputeGraph& graph, const std::vector<ValueRef>& args) { in avg_pool2d() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DPool.cpp228 Tensor avg_pool2d( in avg_pool2d() function
/aosp_15_r20/external/pytorch/benchmarks/tensorexpr/
H A Dpt_engine.py69 def avg_pool2d(self, data, kernel_size, stride=1): member in TorchTensorEngine
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
H A Dpooling.py485 avg_pool2d = average_pooling2d variable
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_caffe2.py259 def avg_pool2d( function
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/
H A Dfunctional.py43 def avg_pool2d( function
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dquantization_patterns.h802 auto avg_pool2d = getInputTensorQParamOpFusionInfo( in quant_fusion_pattern_and_replacements() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dnn_ops.py4579 def avg_pool2d(input, ksize, strides, padding, data_format="NHWC", name=None): # pylint: disable=r… function
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py379 avg_pool2d = _add_docstr( variable
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py4535 def avg_pool2d( function
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dvulkan_api_test.cpp778 TEST_F(VulkanAPITest, avg_pool2d) { in TEST_F() argument