Home
last modified time | relevance | path

Searched defs:avg_pool (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
H A Ddevice_transform_patterns.cc639 auto avg_pool = rewriter.create<TFL::AveragePool2DOp>( in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dnn_ops.py4534 def avg_pool(value, ksize, strides, padding, data_format="NHWC", function
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dmkl_layout_pass.cc855 string avg_pool; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
H A Dtfl_ops.cc3764 auto avg_pool = llvm::dyn_cast<AveragePool2DOp>(op); in GetArithmeticCount() local