Searched defs:to_tensor (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore-src/source/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/mindspore-src/source/tests/st/rewrite/ |
| D | test_sparsify.py | 29 def to_tensor(obj, tensor_type): function
|
| /third_party/mindspore/mindspore-src/source/tests/st/scipy_st/ |
| D | utils.py | 29 def to_tensor(obj, dtype=None, indice_dtype=onp.int32): function
|
| /third_party/mindspore/mindspore-src/source/tests/st/numpy_native/ |
| D | utils.py | 172 def to_tensor(obj, dtype=None): function
|
| /third_party/mindspore/mindspore-src/source/tests/st/networks/models/bert/bert_performance/src/ |
| D | bert_model.py | 455 def construct(self, from_tensor, to_tensor, attention_mask): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/networks/models/bert/src/ |
| D | bert_model.py | 494 def construct(self, from_tensor, to_tensor, attention_mask): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/dataset/vision/ |
| D | py_transforms_util.py | 136 def to_tensor(img, output_type): function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/optimizer/ |
| D | helper.cc | 657 …ShapeValueNode(const FuncGraphPtr &func_graph, const std::vector<int64_t> &shape, bool to_tensor) { in CreateShapeValueNode()
|