Home
last modified time | relevance | path

Searched defs:max_pool2d (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/layers/
H A Dpooling.py38 max_pool2d = max_pooling2d variable
/aosp_15_r20/external/pytorch/aten/src/ATen/native/xnnpack/
H A DMaxPooling.cpp136 Tensor max_pool2d( in max_pool2d() function
H A DShim.cpp80 Tensor max_pool2d( in max_pool2d() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DPooling.cpp140 Tensor max_pool2d( in max_pool2d() function
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dpooling.h223 inline Tensor max_pool2d( in max_pool2d() function
248 inline Tensor max_pool2d( in max_pool2d() function
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DPool.cpp116 void max_pool2d(ComputeGraph& graph, const std::vector<ValueRef>& args) { in max_pool2d() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DPool.cpp246 Tensor max_pool2d( in max_pool2d() function
/aosp_15_r20/external/pytorch/benchmarks/tensorexpr/
H A Dpt_engine.py66 def max_pool2d(self, data, kernel_size, stride=1): member in TorchTensorEngine
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
H A Dpooling.py484 max_pool2d = max_pooling2d variable
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_caffe2.py230 def max_pool2d( function
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/
H A Dfunctional.py508 def max_pool2d( function
/aosp_15_r20/external/pytorch/torch/jit/
H A D_shape_functions.py276 def max_pool2d( function
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dnn_ops.py4970 def max_pool2d(input, ksize, strides, padding, data_format="NHWC", name=None): function
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py833 max_pool2d = boolean_dispatch( variable
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_tracer.py832 def max_pool2d(x): function
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dvulkan_api_test.cpp3418 TEST_F(VulkanAPITest, max_pool2d) { in TEST_F() argument