| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | pooling_ops_3d_test.py | 148 nn_ops.max_pool3d,
158 nn_ops.max_pool3d,
168 nn_ops.max_pool3d,
188 nn_ops.max_pool3d,
197 nn_ops.max_pool3d,
205 nn_ops.max_pool3d,
238 max_pool_3d = nn_ops.max_pool3d(
341 nn_ops.max_pool3d,
351 nn_ops.max_pool3d,
361 nn_ops.max_pool3d,
[all …]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | pooling_ops_3d_test.py | 106 nn_ops.max_pool3d,
116 nn_ops.max_pool3d,
126 nn_ops.max_pool3d,
146 nn_ops.max_pool3d,
155 nn_ops.max_pool3d,
163 nn_ops.max_pool3d,
312 nn_ops.max_pool3d,
322 nn_ops.max_pool3d,
335 nn_ops.max_pool3d,
344 nn_ops.max_pool3d,
[all …]
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | pool3d.py | 89 make_pool3d_tests(tf.nn.max_pool3d)(options, expected_tf_failures=6)
|
| /external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/ |
| D | pool3d.py | 93 make_pool3d_tests(tf.nn.max_pool3d)(options, expected_tf_failures=6)
|
| /external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| D | pooling.h | 297 inline Tensor max_pool3d( in max_pool3d() function
304 return torch::max_pool3d( in max_pool3d()
322 inline Tensor max_pool3d( in max_pool3d() function
325 return detail::max_pool3d( in max_pool3d()
|
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2021-199.md | 1 ## TFSA-2021-199: Crash in `max_pool3d` when size argument is 0 or negative
|
| /external/pytorch/docs/source/ |
| D | nn.functional.rst | 37 max_pool3d
|
| /external/pytorch/aten/src/ATen/native/ |
| D | Pooling.cpp | 167 Tensor max_pool3d( in max_pool3d() function
|
| /external/pytorch/torch/ao/ns/fx/ |
| D | mappings.py | 119 F.max_pool3d,
552 F.max_pool3d,
|
| /external/pytorch/torch/ao/quantization/backend_config/ |
| D | utils.py | 156 elif p is F.max_pool3d:
|
| D | _common_operator_config_utils.py | 646 torch.nn.functional.max_pool3d,
|
| /external/tensorflow/tensorflow/lite/experimental/mlir/testing/ |
| D | build_def.bzl | 48 "max_pool3d",
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 290 csinfo_.max_pool3d = "MaxPool3D"; in MklLayoutRewritePass()
541 {csinfo_.max_pool3d, mkl_op_registry::GetMklOpName(csinfo_.max_pool3d), in MklLayoutRewritePass()
722 {csinfo_.max_pool3d, csinfo_.max_pool3d_grad, 0, 1, 1, 3}); in MklLayoutRewritePass()
898 string max_pool3d; member
1684 mkl_op_registry::GetMklOpName(csinfo_.max_pool3d) && in Maxpool3DGradRewrite()
3713 n->type_string() == csinfo_.max_pool3d || in CheckForNodeRewrite()
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | guard_elimination.cpp | 380 case aten::max_pool3d: in removableGuard()
|
| /external/pytorch/torch/csrc/api/src/nn/modules/ |
| D | pooling.cpp | 122 return F::detail::max_pool3d( in forward()
|
| /external/pytorch/test/nn/ |
| D | test_pooling.py | 469 output, indices = F.max_pool3d(
1452 "max_pool3d": lambda t: F.max_pool3d(t, (5, 1, 1), stride=(5, 1, 1)),
1470 y = torch.nn.functional.max_pool3d(x, 5)
1479 ref_y = torch.nn.functional.max_pool3d(ref_x, 5)
1936 F.max_pool3d(x, kernel_size=(1, 1, 1)).sum().backward()
|
| /external/pytorch/functorch/op_analysis/ |
| D | public_api | 489 nn.functional.max_pool3d
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesDecompositions.cpp | 201 OP_DECOMPOSE(max_pool3d); in TORCH_LIBRARY_IMPL()
|
| /external/pytorch/torch/nn/modules/ |
| D | pooling.py | 296 return F.max_pool3d(
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | pooling.py | 809 nn.max_pool3d,
|
| /external/pytorch/aten/src/ATen/ |
| D | autocast_mode.cpp | 380 KERNEL_CPU(max_pool3d, fp32) in TORCH_LIBRARY_IMPL()
|
| /external/pytorch/test/cpp_api_parity/ |
| D | parity-tracker.md | 171 F::max_pool3d|Yes|No
|
| /external/pytorch/torch/csrc/lazy/ts_backend/ |
| D | ts_native_functions.cpp | 326 at::Tensor LazyNativeFunctions::max_pool3d( in max_pool3d() function in torch::lazy::LazyNativeFunctions
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 1627 ("MAX", 3): max_pool3d, # pylint: disable=undefined-variable
4832 3: gen_nn_ops.max_pool3d
5080 def max_pool3d(input, ksize, strides, padding, data_format="NDHWC", name=None): function
5112 return gen_nn_ops.max_pool3d(
|
| /external/pytorch/torch/ao/quantization/ |
| D | utils.py | 74 torch.nn.functional.max_pool3d,
|