| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | tensor.h | 25 const std::vector<ExprPtr>& reduce_dims, in Tensor()
34 const std::vector<ExprHandle>& reduce_dims, in Tensor()
151 const std::vector<ExprHandle>& reduce_dims) { in Reduce()
201 const std::vector<ExprHandle>& reduce_dims) { in Reduce()
219 const std::vector<ExprHandle>& reduce_dims) { in Reduce()
237 const std::vector<ExprHandle>& reduce_dims) { in Reduce()
250 const std::vector<ExprHandle>& reduce_dims) { in Reduce()
259 const std::vector<ExprHandle>& reduce_dims) { in Reduce()
|
| H A D | tensor.cpp | 12 const std::vector<ExprPtr>& reduce_dims, in constructStmt()
217 const std::vector<ExprHandle>& reduce_dims) { in Reduce()
231 const std::vector<ExprHandle>& reduce_dims) { in Reduce()
241 const std::vector<ExprHandle>& reduce_dims) { in Reduce()
255 const std::vector<ExprHandle>& reduce_dims) { in Reduce()
|
| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_argmax.cpp | 73 array<DenseIndex, 3> reduce_dims; in test_argmax_tuple_reducer() local
105 array<DenseIndex, 3> reduce_dims; in test_argmin_tuple_reducer() local
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | reduce_util.cpp | 39 const auto& reduce_dims = dim_list.value(); in check_dim_list_is_valid() local
268 const auto& reduce_dims = dim_list.value(); in compute_reduced_out_size() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | einsum_spmd_expander.cc | 57 absl::flat_hash_set<std::string> reduce_dims; in ExpandOp() local
389 const Layout& output_layout, absl::flat_hash_set<std::string>& reduce_dims, in MaybeRelayoutInputs()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Normalization.cpp | 289 DimVector reduce_dims(ndim - 1); in batch_norm_cpu_update_stats_template() local
357 DimVector reduce_dims(ndim - 1); in batch_norm_backward_cpu_template() local
828 DimVector reduce_dims(ndim - 1); in batch_norm_cpu() local
946 DimVector reduce_dims(self_sizes.size()); in TORCH_IMPL_FUNC() local
|
| H A D | TensorIteratorReduce.cpp | 140 auto reduce_dims = num_reduce_dims(); in foreach_reduced_elt() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | batch_normalization_test.cc | 529 std::vector<int64_t> reduce_dims; in XLA_TEST_P() local
630 std::vector<int64_t> reduce_dims; in XLA_TEST_P() local
741 std::vector<int64_t> reduce_dims; in XLA_TEST_P() local
|
| H A D | reduce_test.cc | 557 std::vector<int64_t> reduce_dims; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | ExpandUtils.h | 458 c10::SmallVector<int64_t, 8> reduce_dims; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | quantize_and_dequantize_op.h | 236 Eigen::IndexList<Eigen::type2index<0>, Eigen::type2index<2> > reduce_dims; in Compute() local
|
| H A D | fused_batch_norm_op.cc | 174 Eigen::IndexList<Eigen::type2index<0>> reduce_dims; in operator ()() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesConvolution.cpp | 146 DimVector reduce_dims; in compute_grad_bias() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | algebraic_simplifier.cc | 2727 std::vector<int64_t> reduce_dims(dnums.lhs_contracting_dimensions_size()); in HandleDot() local
5421 std::vector<int64_t> reduce_dims = *reduce->mutable_dimensions(); in HandleReduce() local
5540 std::vector<int64_t> reduce_dims; in HandleReduce() local
5746 DimensionVector reduce_dims; in HandleReduceWindow() local
|
| H A D | space_to_batch_converter.cc | 1630 auto reduce_dims = consumer->dimensions(); in SupportedOpForPropagation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | matrix.cc | 273 std::vector<int64_t> reduce_dims; in EinsumDiagonalLabels() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Normalization.cu | 332 DimVector reduce_dims(ndim - 1); in batch_norm_mean_var() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_reductions.py | 1372 def helper(self, shape, reduce_dims, device, dtype): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/ |
| H A D | pass.cc | 1051 tensorflow::BCast::Vec reduce_dims[2]; in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | legalize_hlo.cc | 1885 SmallVector<int64_t, 4> reduce_dims; in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | constant_folding.cc | 680 BCast::Vec reduce_dims[2]; in MaterializeBroadcastGradientArgs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 472 DenseIntElementsAttr reduce_dims, in ApplyReduction()
2263 auto reduce_dims = GetI64ElementsAttr(non_feature_dims, &rewriter); in matchAndRewrite() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 1613 auto reduce_dims = at::DimVector(n); in renorm_backward() local
1647 at::DimVector reduce_dims(self_sizes.size()); in renorm_jvp() local
|