Home
last modified time | relevance | path

Searched refs:TRANSFER_CONCAT_SPLIT_COUNT_INDEX (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dredistribution_operator_infer.cc263 int64_t split_count = args[TRANSFER_CONCAT_SPLIT_COUNT_INDEX]; in TransferConcatByAxis()
Dtensor_redistribution.cc273 double dev_num = attrs[TRANSFER_CONCAT_SPLIT_COUNT_INDEX]; in ComputeConcatCost()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dops_utils.h79 constexpr size_t TRANSFER_CONCAT_SPLIT_COUNT_INDEX = 2; variable