Home
last modified time | relevance | path

Searched defs:tensor_map_index (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dtensordot_info.cc214 void TensorDotInfo::InferTensorMapAxesInt(const TensorMap &tensor_map_index) { in InferTensorMapAxesInt()
233 const TensorMap &tensor_map_index) { in InferTensorMapAxesTuple()
284 TensorMap tensor_map_index, input_a_tensor_map; in InferTensorMap() local
Dbatch_parallel_info.cc87 Shape tensor_map_index; in InferTensorMap() local
98 Shape tensor_map_index; in InferTensorMap() local
Dtmp_identity_info.cc39 Shape tensor_map_index; in InferTensorMap() local
Darithmetic_info.cc114 TensorMap tensor_map_index; in SetExpendTensorMap() local
153 Shape tensor_map_index; in InferTensorMap() local
Dreluv2_info.cc82 Shape tensor_map_index; in InferTensorMap() local
Dloss_info.cc80 Shape tensor_map_index; in InferTensorMap() local
Dvirtual_dataset_info.cc105 Shape tensor_map_index; in InferTensorMap() local
Dget_next_info.cc40 Shape tensor_map_index; in InferTensorMap() local
Ddropout_do_mask_info.cc80 Shape tensor_map_index; in InferTensorMap() local
Dreshape_info.cc297 Shape tensor_map_index; in InferDefaultLayout() local
Dactivation_info.cc246 Shape tensor_map_index; in InferTensorMap() local
Dmatmul_info.cc255 Shape tensor_map_index; in InferTensorMap() local
Dreduce_method_info.cc111 Shape tensor_map_index, output_tensor_map; in InferTensorMap() local
Dgather_v2_p_info.cc469 Shape tensor_map_index; in InferInputsTensorMap() local