Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dgather_v2_info.cc188 …if ((input_tensor_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_.at(0), input_shape) … in InferTensorInfo()
189 …(input_index_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_.at(1), input_index_shape)… in InferTensorInfo()
190 …(output_tensor_layout.InitFromVector(dev_matrix_shape_, outputs_tensor_map_.at(0), output_shape) !… in InferTensorInfo()
Dgatherd_info.cc120 …if (input_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_[i], inputs_shape_[i]) != SUC… in InferTensorInfo()
133 …if (output_layout.InitFromVector(dev_matrix_shape_, outputs_tensor_map_[i], outputs_shape_[i]) != … in InferTensorInfo()
Dreduce_method_info.cc365 …if ((input_tensor_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_[0], input_shape) != … in InferTensorInfo()
366 …(output_tensor_layout.InitFromVector(dev_matrix_shape_, outputs_tensor_map_[0], output_shape) != S… in InferTensorInfo()
494 …if ((input_tensor_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_[0], input_shape) != … in InferTensorInfo()
495 …(output_tensor_layout.InitFromVector(dev_matrix_shape_, outputs_tensor_map_[0], output_shape) != S… in InferTensorInfo()
Dmatmul_info.cc327 …if ((mat_a_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_[0], inputs_shape_[0]) != SU… in InferTensorLayout()
328 …(mat_b_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_[1], inputs_shape_[1]) != SUCCES… in InferTensorLayout()
329 …(output_layout.InitFromVector(output_dev_matrix_shape, outputs_tensor_map_[0], outputs_shape_[0]) … in InferTensorLayout()
Dgather_v2_p_info.cc575 …if ((input_tensor_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_.at(0), input_shape) … in InferTensorInfo()
576 …(input_index_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_.at(1), input_index_shape)… in InferTensorInfo()
577 …(output_tensor_layout.InitFromVector(out_dev_matrix_shape_, outputs_tensor_map_.at(0), output_shap… in InferTensorInfo()
Dget_next_info.cc63 …if (output_layout.InitFromVector(dev_matrix_shape_, outputs_tensor_map_[i], outputs_shape_[i]) != … in InferTensorLayout()
Dreshape_info.cc301 Status status = layout->InitFromVector({dev_num_}, tensor_map_index, shape); in InferDefaultLayout()
Doperator_info.cc210 …if (input_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_[i], inputs_shape_[i]) != SUC… in InferTensorInfo()
220 …if (output_layout.InitFromVector(dev_matrix_shape_, outputs_tensor_map_[i], outputs_shape_[i]) != … in InferTensorInfo()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dtensor_layout.h43 …Status InitFromVector(const Shape &device_arrangement, const Shape &tensor_map, const Shape &tenso…
Dtensor_layout.cc71 Status TensorLayout::InitFromVector(const Shape &device_arrangement, const Shape &tensor_map, in InitFromVector() function in mindspore::parallel::TensorLayout
413 repeat.InitFromVector(dev_mat, tensor_map, tensor_shape); in TransferRepeatLayout()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dparameter_manager.cc601 if (new_tensor_layout.InitFromVector(dev_mat, tensor_map, origin_shape) != SUCCESS) { in HandleAdaFactorOpt()
Dstep_parallel.cc2095 …if (input_tensor_layout.InitFromVector(dev_matrix_array, input_tensor_map_array, input_shape_array… in CreateParameterLayout()
2121 …if (stand_alone_layout.InitFromVector(dev_matrix_array, stand_alone_tensor_map_array, input_shape_… in InferSensRedistribution()