Searched refs:TensorNames (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/aten/src/ATen/ |
| D | TensorNames.cpp | 48 TensorNames::TensorNames(ArrayRef<Dimname> names) { in TensorNames() function in at::namedinference::TensorNames
55 TensorNames::TensorNames(ArrayRef<Dimname> names, int64_t start, int64_t end) { in TensorNames() function in at::namedinference::TensorNames
65 TensorNames& TensorNames::unifyFromRightInplace(const TensorNames& other, const char* op_name) { in unifyFromRightInplace()
87 void TensorNames::append(TensorName name) { in append()
91 void TensorNames::checkUnique(const char* op_name) const { in checkUnique()
120 std::vector<Dimname> TensorNames::toDimnameVec() const { in toDimnameVec()
|
| D | TensorNames.h | 52 struct TORCH_API TensorNames { struct
53 explicit TensorNames(ArrayRef<Dimname> names);
58 explicit TensorNames(ArrayRef<Dimname> names, int64_t start, int64_t end);
61 TensorNames& unifyFromRightInplace(
62 const TensorNames& other,
70 explicit TensorNames(TensorNameVec&& names) : names_(std::move(names)){}; in TensorNames() function
|
| D | NamedTensorUtils.cpp | 327 auto wrapped_self_names = TensorNames(self_names, 0, num_batch_dims(self_names)); in compute_matmul_outnames()
328 const auto wrapped_other_names = TensorNames(other_names, 0, num_batch_dims(other_names)); in compute_matmul_outnames()
462 auto self_batch = TensorNames(self_names, 0, num_batch_dims(self_names)); in compute_cdist_outnames()
463 const auto other_batch = TensorNames(other_names, 0, num_batch_dims(other_names)); in compute_cdist_outnames()
|
| /external/pytorch/aten/src/ATen/test/ |
| D | NamedTensor_test.cpp | 13 using at::namedinference::TensorNames;
126 auto names_wrapper = at::namedinference::TensorNames(names); in tensornames_unify_from_right()
127 auto other_wrapper = at::namedinference::TensorNames(other_names); in tensornames_unify_from_right()
230 TensorNames(names).checkUnique("op_name"); in TEST()
234 auto tensornames = TensorNames(nchh); in TEST()
|
| /external/pytorch/ |
| D | build_variables.bzl | 1028 "aten/src/ATen/TensorNames.cpp",
|