/aosp_15_r20/external/tensorflow/tensorflow/python/layers/ |
H A D | pooling.py | 39 avg_pool2d = average_pooling2d variable
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
H A D | AveragePool2d.cpp | 17 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 D | pooling.h | 53 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 D | Pool.cpp | 192 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 D | Pool.cpp | 228 Tensor avg_pool2d( in avg_pool2d() function
|
/aosp_15_r20/external/pytorch/benchmarks/tensorexpr/ |
H A D | pt_engine.py | 69 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 D | pooling.py | 485 avg_pool2d = average_pooling2d variable
|
/aosp_15_r20/external/pytorch/torch/onnx/ |
H A D | symbolic_caffe2.py | 259 def avg_pool2d( function
|
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/ |
H A D | functional.py | 43 def avg_pool2d( function
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
H A D | quantization_patterns.h | 802 auto avg_pool2d = getInputTensorQParamOpFusionInfo( in quant_fusion_pattern_and_replacements() local
|
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
H A D | nn_ops.py | 4579 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 D | functional.py | 379 avg_pool2d = _add_docstr( variable
|
/aosp_15_r20/external/pytorch/torch/_inductor/ |
H A D | lowering.py | 4535 def avg_pool2d( function
|
/aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
H A D | vulkan_api_test.cpp | 778 TEST_F(VulkanAPITest, avg_pool2d) { in TEST_F() argument
|