Searched defs:set_input_shape (Results 1 – 2 of 2) sorted by relevance
59 void set_input_shape(const std::vector<int> &shape) { input_shape_ = shape; } in set_input_shape() function
122 void GroupConv2DGradInput::set_input_shape(const std::vector<int64_t> &input_shape) { in set_input_shape() function in mindspore::ops::GroupConv2DGradInput