Home
last modified time | relevance | path

Searched defs:input_tensor_list (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/execute/
Dbindings.cc39 … [](PyExecute &self, const std::vector<std::shared_ptr<Tensor>> &input_tensor_list) { in __anona55c35fb0102()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/
Dtensorlist_c_utils.c77 TensorListC *input_tensor_list = (TensorListC *)tensorc; in InferFlagTensorList() local
/third_party/mindspore/mindspore-src/source/tests/st/nontask_sink/
Dtest_all_to_all_v.py40 def construct(self, output_tensor_list, input_tensor_list, group=None): argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/communication/
Dcomm_func.py920 def all_to_all_with_output_shape(output_shape_list, input_tensor_list, group=None): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/api/
Dexecute.cc354 Status Execute::operator()(const std::vector<MSTensor> &input_tensor_list, std::vector<MSTensor> *o… in operator ()()
436 Status PyExecute::operator()(const std::vector<std::shared_ptr<Tensor>> &input_tensor_list, in operator ()()
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/dataset/transforms/
Dc_transforms.py38 def __call__(self, *input_tensor_list):
Dtransforms.py73 def __call__(self, *input_tensor_list):
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/common/
Dtensor_util.cc711 auto input_tensor_list = reinterpret_cast<const TensorList *>(input); in IsUnKnownDtype() local
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/dataset/vision/
Dtransforms.py92 def __call__(self, *input_tensor_list):
109 def __call__(self, *input_tensor_list):
Dc_transforms.py67 def __call__(self, *input_tensor_list):
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/dataset/audio/
Dtransforms.py49 def __call__(self, *input_tensor_list):