Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dredistribution_operator_infer.cc261 int64_t tensor_dim = args[TRANSFER_CONCAT_TENSOR_DIM_INDEX]; in TransferConcatByAxis()
Dtensor_redistribution.cc278 int64_t concat_dim = attrs[TRANSFER_CONCAT_TENSOR_DIM_INDEX]; in ComputeConcatCost()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dops_utils.h77 constexpr size_t TRANSFER_CONCAT_TENSOR_DIM_INDEX = 0; variable