Searched refs:UpdateTensorMap (Results 1 – 4 of 4) sorted by relevance
226 …if (cur_tensor_layout_.UpdateTensorMap(index, args[TRANSFER_PERMUTE_CONCAT_DIM_INDEX]) == Status::… in TransferSplitByAxis()247 if (cur_tensor_layout_.UpdateTensorMap(LongToSize(val), NONE) == Status::FAILED) { in TransferPermuteByAxis()250 if (cur_tensor_layout_.UpdateTensorMap(index, out_dim) == Status::FAILED) { in TransferPermuteByAxis()284 if (cur_tensor_layout_.UpdateTensorMap(LongToSize(tensor_dim), NONE) == Status::FAILED) { in TransferConcatByAxis()
93 Status UpdateTensorMap(size_t index, int64_t value);
343 Status TensorLayout::UpdateTensorMap(size_t index, int64_t value) { in UpdateTensorMap() function in mindspore::parallel::TensorLayout
234 TEST_F(TestTensorLayout, UpdateTensorMap) { in TEST_F() argument246 tensor_layout.UpdateTensorMap(1, 0); in TEST_F()