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