| /aosp_15_r20/external/executorch/backends/cadence/hifi/operators/ |
| H A D | quantized_linear_out.cpp | 46 const int64_t in_dim = weight.size(1); // = in_dim in _quantized_linear_asym8u() local
91 const int64_t in_dim = weight.size(1); // = in_dim in _quantized_linear_asym8s() local
136 const int64_t in_dim = weight.size(1); // = in_dim in _quantized_linear_per_tensor_asym8u() local
184 const int64_t in_dim = weight.size(1); // = in_dim in _quantized_linear_per_tensor_asym8s() local
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | reduce_util.cpp | 25 inline size_t _normalize_non_neg_d(ssize_t d, ssize_t in_dim) { in _normalize_non_neg_d()
107 const size_t in_dim = in.dim(); in get_reduced_dim_product() local
227 const auto in_dim = in.dim(); in compute_reduced_out_size() local
264 const auto in_dim = in.dim(); in compute_reduced_out_size() local
|
| H A D | kernel_ops_util.h | 157 size_t in_dim = in_sizes.size(); in kernel_reduction_then_map_2d() local
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | Select.cpp | 28 const int64_t in_dim = t_in.dim(); in check_args() local
72 int64_t in_dim = t_in->dim(); in add_select_int_node() local
|
| H A D | Repeat.cpp | 34 int64_t in_dim = in.dim(); in check_args() local
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_vmap.py | 3345 def test_chunk_vmap(self, in_dim, out_dim, randomness): argument
3371 def test_vmap_chunksize(self, in_dim, out_dim, randomness): argument
3465 def test_vmap_chunksize_error(self, in_dim, out_dim, randomness): argument
3501 def test_vmap_chunksize_composition(self, in_dim, out_dim, randomness): argument
5000 def append(inp, in_dim): argument
6014 def test_chunk_vmap(self, in_dim, out_dim): argument
6035 def test_vmap_chunksize(self, in_dim, out_dim): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | tensor.cc | 1351 const int64_t in_dim = out_dim + offset; in ComputeFlatInnerDims() local
1354 for (int64_t in_dim = 0; in_dim < offset; ++in_dim) { in ComputeFlatInnerDims() local
1366 for (int64_t in_dim = num_out_dims; in_dim < orig.size(); ++in_dim) { in ComputeFlatOuterDims() local
|
| /aosp_15_r20/external/executorch/backends/cadence/reference/operators/ |
| H A D | quantized_matmul_out.cpp | 72 size_t in_dim = X.size(X.dim() - 1); in _typed_quantized_matmul() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Concat.cpp | 272 Tensor cat(const at::ITensorListRef& tensors, const int64_t in_dim) { in cat()
|
| /aosp_15_r20/external/pytorch/torch/backends/_nnapi/ |
| H A D | serializer.py | 601 def forward_operand_shape(self, out_op_id, out_dim, in_op_id, in_dim): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | array_ops.cc | 497 auto in_dim = c->Dim(in, in_rank - i - 1); in __anon38bbb0e80702() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | eigen_spatial_convolutions_test.cc | 1327 DSizes<ptrdiff_t, 4> in_dim; in TEST() local
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_repros.py | 1657 def fcnn(in_dim, out_dim, hidden_dim, activation=torch.nn.GELU): argument
|