| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | op_permute_copy_test.cpp | 36 const std::vector<int64_t> new_dim = {0}; in TEST_F() local
51 const std::vector<int64_t> new_dim = {1, 0, 2}; in TEST_F() local
78 const std::vector<int64_t> new_dim = {1, 0}; in TEST_F() local
106 const std::vector<int64_t> new_dim = {2, 0, 1}; in TEST_F() local
140 const std::vector<int64_t> new_dim = {0, 3, 2, 1}; in TEST_F() local
227 const std::vector<int64_t> new_dim = {4, 3, 2, 1, 0}; in TEST_F() local
307 const std::vector<int64_t> new_dim = {4, 3, 2, 1, 0}; in TEST_F() local
322 const std::vector<int64_t> new_dim = {0, 1, 1}; in TEST_F() local
338 const std::vector<int64_t> new_dim = {1, 1, 1}; in TEST_F() local
354 const std::vector<int64_t> new_dim = {0, 1, -2}; in TEST_F() local
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ops/ |
| H A D | tpu_partitioned_output_op.cc | 45 shape_inference::DimensionHandle new_dim; in __anon0352b5690102() local
|
| H A D | tpu_partitioned_input_op.cc | 29 shape_inference::DimensionHandle new_dim; in _UpdatePartitionDim() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| H A D | conv_emitter_transforms.cc | 121 unsigned new_dim = affine_map.operands.size(); in TileLoop() local
|
| /aosp_15_r20/external/pytorch/functorch/dim/ |
| H A D | __init__.py | 77 def cat(tensors, dim, new_dim): argument
|
| H A D | reference.py | 578 def stack(tensors, new_dim, dim=0, out=None): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | PadNd.cpp | 74 auto new_dim = input_sizes[l_diff + i] + pad[pad_idx] + pad[pad_idx + 1]; in constant_pad_nd() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | runtime_shape.h | 197 const int new_dim = in ToRuntimeDims() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_conv_padding_legalization.cc | 248 auto* new_dim = new_backward_conv_window.mutable_dimensions(i); in CanonicalizeBackwardFilterConvolution() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | algebraic_simplifier.cc | 1979 int64_t new_dim = lhs_dimension_map[dim]; in RemoveDegenerateDimensionFromDot() local
1985 int64_t new_dim = lhs_dimension_map[dim]; in RemoveDegenerateDimensionFromDot() local
1992 int64_t new_dim = rhs_dimension_map[dim]; in RemoveDegenerateDimensionFromDot() local
1998 int64_t new_dim = rhs_dimension_map[dim]; in RemoveDegenerateDimensionFromDot() local
6463 auto new_dim = swapped_window.add_dimensions(); in SwapConvOperands() local
|
| H A D | convolution_group_converter.cc | 537 auto new_dim = new_window.add_dimensions(); in HandleConvolution() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| H A D | xla_ops.cc | 544 int64_t new_dim = orig_dim + low + high; in __anon6b9e6b800302() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | LegacyBatchingRegistrations.cpp | 575 auto new_dim = bdim_size.has_value() ? dim + 1 : dim; in cat_batching_rule() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | shape_inference.cc | 661 DimensionHandle new_dim, ShapeHandle* out) { in ReplaceDim()
|
| H A D | common_shape_fns.cc | 235 DimensionHandle new_dim = c->RankKnown(input_shape) in EinsumShape() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | shape_inference.cpp | 242 auto new_dim = input_sizes[l_diff + i] + pad[pad_idx] + pad[pad_idx + 1]; in compute_shape_constant_pad_nd() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | graph_properties.cc | 1195 DimensionHandle new_dim = GetUnknownOutputDim(node, port_index, d); in OutputAsUnion() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | lowerings.cpp | 1805 auto new_dim = at::maybe_wrap_dim(permute_dims[i], A.ndim()); in nnc_lowerings_lazy_registration() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | dot_handler.cc | 192 WindowDimension* new_dim = new_window.add_dimensions(); in GenNewWindow() local
|
| /aosp_15_r20/external/pytorch/functorch/csrc/dim/ |
| H A D | dim.cpp | 2715 mpy::handle tensors, new_dim, dim; in py_stack() local
|