Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc2085 Shape input_shape_array = inputs_shape[0]; in CreateParameterLayout() local
2086 if (input_shape_array.empty()) { in CreateParameterLayout()
2090 size_t shape_size = input_shape_array.size(); in CreateParameterLayout()
2095 …sor_layout.InitFromVector(dev_matrix_array, input_tensor_map_array, input_shape_array) != SUCCESS)… in CreateParameterLayout()
2112 Shape input_shape_array = inputs_shape[0]; in InferSensRedistribution() local
2113 if (input_shape_array.empty()) { in InferSensRedistribution()
2118 TensorMap stand_alone_tensor_map_array(SizeToLong(input_shape_array.size()), -1); in InferSensRedistribution()
2121 …yout.InitFromVector(dev_matrix_array, stand_alone_tensor_map_array, input_shape_array) == FAILED) { in InferSensRedistribution()