Home
last modified time | relevance | path

Searched defs:input_a_tensor_map (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dtensordot_info.cc232 void TensorDotInfo::InferTensorMapAxesTuple(size_t size, const TensorMap &input_a_tensor_map, in InferTensorMapAxesTuple()
284 TensorMap tensor_map_index, input_a_tensor_map; in InferTensorMap() local
Dgather_v2_p_info.cc383 Shape input_a_tensor_map = inputs_tensor_map_.at(0); in InferMirrorOps() local