Searched defs:set_block_shape (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ | ||
| D | space_to_batch_nd.cc | 82 void SpaceToBatchND::set_block_shape(std::vector<int64_t> block_shape) { in set_block_shape() function in mindspore::ops::SpaceToBatchND |
| D | batch_to_space_nd.cc | 84 void BatchToSpaceND::set_block_shape(std::vector<int64_t> block_shape) { in set_block_shape() function in mindspore::ops::BatchToSpaceND |