| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | bincount_op.cc | 78 Eigen::array<int, 1> reduce_dim({0}); in Compute() local
161 Eigen::array<int, 1> reduce_dim({0}); in Compute() local
|
| H A D | reduce_join_op.cc | 69 for (int32_t reduce_dim : reduced_indices) { in GetReductionIterSize() local
|
| H A D | tile_ops.cc | 492 Eigen::DSizes<Eigen::DenseIndex, REDUCENDIM> reduce_dim; in HandleReduce() local
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | Reduce.cpp | 47 int32_t reduce_dim = dim; in add_reduce_node() local
|
| H A D | Softmax.cpp | 44 int32_t reduce_dim = graph.extract_scalar<int32_t>(dim); in add_softmax_node() local
|
| /aosp_15_r20/external/pytorch/torch/masked/maskedtensor/ |
| H A D | reductions.py | 84 def reduce_dim(self, dim, keepdim=False, dtype=None): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/strings_ops/ |
| H A D | reduce_join_op_test.py | 43 def _joined_array(num_dims, reduce_dim): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | matmul_spmd_expander.cc | 68 std::string reduce_dim; in ExpandOp() local
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset13.py | 431 def reduce_dim(g, self, dim, keepdim, dtype): function
|
| H A D | symbolic_helper.py | 1800 def reduce_dim(g, self, dim, keepdim, dtype): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | matrix.cc | 93 int64_t reduce_dim = n_dims - 1; in GetMatrixDiagonal() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | lower_tf.cc | 1658 auto reduce_dim = in matchAndRewrite() local
|
| H A D | legalize_hlo.cc | 1507 auto reduce_dim = (*dimensions.value_begin<APInt>()).getSExtValue(); in MatchIotaConst() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorConversions.cpp | 1056 std::vector<int64_t> reduce_dim((blocked_layout ? 2 : 0) + dense_dim); in dense_to_sparse_compressed() local
|
| H A D | LinearAlgebra.cpp | 2822 auto reduce_dim = is_all_reduce ? all_dim : opt_dim_; in TORCH_IMPL_FUNC() local
|
| /aosp_15_r20/external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/ |
| H A D | onedevice_train_palm_detection_filter_inference_beta.cc | 823 const int32_t reduce_dim, in Softmax()
|
| H A D | onedevice_train_palm_detection_filter_inference.cc | 824 const int32_t reduce_dim, in Softmax()
|
| H A D | onedevice_train_palm_detection_filter_inference_v2.cc | 823 const int32_t reduce_dim, in Softmax()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | randomized_tests.cc | 1684 int reduce_dim = in TEST_F() local
1701 int reduce_dim = in TEST_F() local
|