Searched defs:get_block_shape (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ | ||
| D | batch_to_space_nd.cc | 138 std::vector<int64_t> BatchToSpaceND::get_block_shape() const { in get_block_shape() function in mindspore::ops::BatchToSpaceND |
| D | space_to_batch_nd.cc | 184 std::vector<int64_t> SpaceToBatchND::get_block_shape() const { in get_block_shape() function in mindspore::ops::SpaceToBatchND |