| /external/pytorch/aten/src/ATen/core/ |
| D | NamedTensor.cpp | 55 void check_names_valid_for(size_t tensor_dim, DimnameList names) { in check_names_valid_for() argument
57 tensor_dim <= kMaxNamedTensorDim, in check_names_valid_for()
59 "Attempted to create a tensor with dim ", tensor_dim, " with names ", names); in check_names_valid_for()
60 TORCH_CHECK(tensor_dim == names.size(), in check_names_valid_for()
62 "number of dimensions in tensor (", tensor_dim, ") ", in check_names_valid_for()
|
| D | NamedTensor.h | 100 void check_names_valid_for(size_t tensor_dim, DimnameList names);
|
| /external/pytorch/aten/src/ATen/native/mkldnn/xpu/detail/ |
| D | Utils.cpp | 217 auto tensor_dim = sizes.size(); in is_onednn_matmul_strides() local
218 if (tensor_dim != 2 && tensor_dim != 3) in is_onednn_matmul_strides()
227 for (size_t dim = 0; dim < tensor_dim; ++dim) in is_onednn_matmul_strides()
245 if (strides[tensor_dim - 1] != 1 && strides[tensor_dim - 2] != 1) in is_onednn_matmul_strides()
|
| /external/pytorch/aten/src/ATen/native/nested/ |
| D | NestedTensorUtils.cpp | 93 int64_t tensor_dim = sizes.size(1); in chunk_nested_tensor() local
104 const int64_t index = i * tensor_dim + dim; in chunk_nested_tensor()
148 int64_t tensor_dim = sizes.size(1); in split_with_sizes_nested() local
159 const int64_t index = i * tensor_dim + dim; in split_with_sizes_nested()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | NamedTensor.cpp | 196 const auto tensor_dim = tensor.sizes().size(); in align_to() local
232 const auto num_ellipsis_names = countUnset(order_has_tensor_name, tensor_dim); in align_to()
241 auto setNewSizesStridesNamesFor = [&](int64_t out_dim, int64_t tensor_dim) { in align_to() argument
242 new_sizes[out_dim] = tensor_sizes[tensor_dim]; in align_to()
243 new_strides[out_dim] = tensor_strides[tensor_dim]; in align_to()
244 new_names[out_dim] = tensor_names[tensor_dim]; in align_to()
263 for (const auto tensor_idx : c10::irange(tensor_dim)) { in align_to()
|
| /external/pytorch/aten/src/ATen/ |
| D | LegacyBatchedTensorImpl.cpp | 118 auto tensor_dim = tensor.dim(); in makeBatched() local
120 tensor_dim <= kVmapMaxTensorDims, in makeBatched()
122 "; got a tensor with dim ", tensor_dim); in makeBatched()
|
| D | ExpandUtils.cpp | 67 int64_t tensor_dim = static_cast<int64_t>(tensor_sizes.size()); in inferExpandGeometryImpl() local
69 if (tensor_dim == 0) { in inferExpandGeometryImpl()
80 int64_t dim = tensor_dim - 1 - offset; in inferExpandGeometryImpl()
|
| D | LegacyVmapTransforms.cpp | 169 int64_t tensor_dim = 0; in alignBatchDimsAtFront() local
174 aligned_sizes[bdim] = physical_sizes[tensor_dim++]; in alignBatchDimsAtFront()
|
| /external/pytorch/torch/_inductor/codegen/ |
| D | triton_split_scan.py | 61 tensor_dim = 0 if is_reduction else None
72 tensor_dim=tensor_dim,
|
| D | simd.py | 122 tensor_dim: Optional[int],
149 self.tensor_dim = tensor_dim
377 tensor_dim = tensor_dims.find(prefix) if prefix in tensor_dims else None
389 tensor_dim=tensor_dim,
420 return sum(int(tree.tensor_dim is not None) for tree in self.range_trees)
430 if tree.tensor_dim is None:
434 sizes[tree.tensor_dim] = f"{tree.prefix.upper()}BLOCK"
|
| D | triton.py | 2698 if tree.tensor_dim is None:
2871 assert entry.tensor_dim is not None
2872 size = self.indexing_size_str(entry.tensor_dim)
2938 if entry.tensor_dim is not None:
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | check_ops.py | 1970 tensor_dim = i - len(sizes.symbolic_sizes)
1972 tensor_dim = i
1990 actual_size = sizes.actual_sizes[tensor_dim]
1997 tensor_dim, specified_size, actual_size,
2009 'Tensor %s dimension' % tensor_name(sizes.x), tensor_dim,
2019 size = sizes.actual_sizes[tensor_dim]
2020 size_specifications[size_symbol] = (size, sizes.x, tensor_dim)
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | symbolic_shape_runtime_fusion.cpp | 685 const int64_t tensor_dim = sizes[dim_index]; in __anon74f36cf50602() local
687 if (C10_UNLIKELY(dim_value != tensor_dim)) { in __anon74f36cf50602()
699 if (C10_UNLIKELY(flattened_sym_value != tensor_dim)) { in __anon74f36cf50602()
705 flattened_symbolic_dims[flattened_sym_index] = tensor_dim; in __anon74f36cf50602()
|
| /external/tensorflow/tensorflow/dtensor/python/ |
| D | layout.py | 579 tensor_dim = layout_proto.sharding_specs.add()
580 tensor_dim.sharding_spec = dim_sharding
|
| /external/pytorch/torch/jit/ |
| D | _shape_functions.py | 107 tensor_dim = len(self)
113 dim = tensor_dim - 1 - offset
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | dtensor_device.cc | 768 int64_t tensor_dim = TFE_TensorHandleDim(component.get(), i, status); in VerifyPackTensorShapeAndDtype() local
770 if (tensor_dim != (*component_shape)[i]) { in VerifyPackTensorShapeAndDtype()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 1901 const auto tensor_dim = tensor->getDimensions(); in ConvertConv2DHelper() local
1902 const int c_dim_size = tensor_dim.d[params->use_implicit_batch ? 0 : 1]; in ConvertConv2DHelper()
2859 const auto tensor_dim = tensor->getDimensions(); in ConvertConv3DHelper() local
2914 const int num_groups = (group == 0) ? tensor_dim.d[0] : group; in ConvertConv3DHelper()
|