Searched defs:x_input_shape (Results 1 – 2 of 2) sorted by relevance
49 …auto x_input_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->Get… in LSTMInferShape() local
3307 auto x_input_shape = dyn_cast<abstract::Shape>(node->input(kOneNum)->Shape())->shape(); in ExportPrimDynamicRNN() local3507 auto x_input_shape = dyn_cast<abstract::Shape>(node->input(kOneNum)->Shape())->shape(); in ExportPrimLSTM() local3787 auto x_input_shape = dyn_cast<abstract::Shape>(x_input->Shape())->shape(); in ExportPrimSort() local