Searched defs:to_tensor (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/tests/mindspore_test_framework/apps/ |
| D | bert_attention_submodules.py | 71 def construct(self, from_tensor, to_tensor): argument
263 def construct(self, to_tensor, attention_scores): argument
|
| /third_party/mindspore/tests/st/numpy_native/ |
| D | utils.py | 175 def to_tensor(obj, dtype=None): function
|
| /third_party/mindspore/tests/st/networks/models/bert/src/ |
| D | bert_model.py | 494 def construct(self, from_tensor, to_tensor, attention_mask): argument
|
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | model.py | 445 def construct(self, from_tensor, to_tensor, attention_mask): argument
|
| /third_party/mindspore/mindspore/dataset/vision/ |
| D | py_transforms_util.py | 131 def to_tensor(img, output_type): function
|
| /third_party/mindspore/mindspore/common/ |
| D | tensor.py | 1272 def to_tensor(self, slice_index=None, shape=None, opt_shard_group=None): member in Tensor
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | helper.cc | 504 …ShapeValueNode(const FuncGraphPtr &func_graph, const std::vector<int64_t> &shape, bool to_tensor) { in CreateShapeValueNode()
|