Searched defs:tensor_num (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/ |
| D | elewise_calculation_ops_proto.cc | 126 int64_t tensor_num; in GetAddNConstValue() local
134 int64_t AddNInferClassify(ge::Operator &op, int64_t &tensor_num) { in AddNInferClassify()
182 int64_t tensor_num = GetAddNConstValue(op); in IMPLEMT_COMMON_INFERFUNC() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/providers/dpico/infer/ |
| D | custom_infer.cc | 40 …e(const std::map<std::string, std::string> &attrs, const std::string &attr_name, size_t tensor_num, in GetCustomShape()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/infer/ |
| D | dpico_custom_infer.cc | 77 …e(const std::map<std::string, std::string> &attrs, const std::string &attr_name, size_t tensor_num, in GetCustomShape()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/providers/nnie/ |
| D | nnie_micro.cc | 47 int MakeTensorList(TensorC *tensors, int tensor_num, NnieTensors *tensor_list) { in MakeTensorList()
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_dynamic_ops.py | 78 def get_columns(tensor_num): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/profiler/parser/ |
| D | framework_parser.py | 128 def _special_process_tensor_data(item_binary_data, data_type, tensor_num): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/distributed/persistent/storage/ |
| D | local_file.cc | 171 size_t tensor_num = inputs.size(); in WriteBlockFiles() local
|