Searched refs:max_pool_with_argmax (Results 1 – 13 of 13) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/perception/ |
| H A D | max_pool_with_argmax.cc | 31 namespace max_pool_with_argmax { namespace
245 max_pool_with_argmax::Init, max_pool_with_argmax::Free, in RegisterMaxPoolWithArgmax()
246 max_pool_with_argmax::Prepare, max_pool_with_argmax::Eval}; in RegisterMaxPoolWithArgmax()
|
| H A D | BUILD | 18 "max_pool_with_argmax.cc",
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| H A D | pooling_ops_test.py | 914 out_op, _ = nn_ops.max_pool_with_argmax(t, ksize, strides, padding)
934 _, argmax_op = nn_ops.max_pool_with_argmax(t, ksize, strides, padding)
966 _, argmax_op = nn_ops.max_pool_with_argmax(t, ksize, strides, padding)
1007 out_op, argmax_op = nn_ops.max_pool_with_argmax(
2282 p, am = nn_ops.max_pool_with_argmax(
2292 nn_ops.max_pool, nn_ops.avg_pool, nn_ops.max_pool_with_argmax
2307 pool_funcs.append(nn_ops.max_pool_with_argmax)
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/testing/op_tests/ |
| H A D | max_pool_with_argmax.py | 44 updates, indices = tf.nn.max_pool_with_argmax(
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/ |
| H A D | max_pool_with_argmax.py | 48 updates, indices = tf.nn.max_pool_with_argmax(
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/mlir/testing/ |
| H A D | build_def.bzl | 49 "max_pool_with_argmax",
|
| H A D | generate_examples.py | 51 from tensorflow.lite.experimental.mlir.testing.op_tests.max_pool_with_argmax import make_max_pool_w…
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/testing/ |
| H A D | generate_examples_lib.py | 113 from tensorflow.lite.testing.op_tests.max_pool_with_argmax import make_max_pool_with_argmax_tests
|
| H A D | build_def.bzl | 97 "max_pool_with_argmax",
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | nn_ops.py | 5185 return gen_nn_ops.max_pool_with_argmax(
5215 return gen_nn_ops.max_pool_with_argmax(
5225 max_pool_with_argmax_v1.__doc__ = gen_nn_ops.max_pool_with_argmax.__doc__
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
| H A D | tensorflow.nn.pbtxt | 244 name: "max_pool_with_argmax"
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
| H A D | tensorflow.nn.pbtxt | 288 name: "max_pool_with_argmax"
|
| /aosp_15_r20/external/tensorflow/ |
| H A D | RELEASE.md | 636 * (since v2.7) `tf.nn.max_pool_with_argmax` gradient
1477 * `tf.nn.max_pool_with_argmax` gradient
|