Searched refs:block_shape (Results 1 – 11 of 11) sorted by relevance
6 将一个4维tensor的batch维度按block_shape切分成小块，并将这些小块拼接到空间维度。18 - $ n' = n / (block_shape[0] \* block_shape[1]) $19 - $ h' = h \* block_shape[0] - crops[0][0] - crops[0][1] $20 - $ w' = w \* block_shape[1] - crops[1][0] - crops[1][1] $
34 | struct [BatchToSpaceND](_batch_to_space_n_d_v10.md) | 将一个4维tensor的batch维度按block_shape切…
34 | struct [BatchToSpaceND](_batch_to_space_n_d_v20.md) | 将一个4维tensor的batch维度按block_shape切…
43 | struct [BatchToSpaceND](_batch_to_space_n_d_v10.md) | 将一个4维tensor的batch维度按block_shape切…
43 | struct [BatchToSpaceND](_batch_to_space_n_d_v20.md) | 将一个4维tensor的batch维度按block_shape切…
65 block_shape.push_back(blockShapeData[i]); in SetBlockShape()181 auto primitive = mindspore::lite::MindIR_SpaceToBatchND_CreatePrimitive(block_shape, paddings); in GetPrimitive()
45 std::vector<int64_t> block_shape {};
1969 std::vector<int64_t> block_shape {}; variable1970 void* primitive = mindspore::lite::MindIR_SpaceToBatchND_CreatePrimitive(block_shape, paddings);
2887 std::vector<int64_t> block_shape {}; variable2888 void* primitive = mindspore::lite::MindIR_SpaceToBatchND_CreatePrimitive(block_shape, paddings);
2826 std::vector<int64_t> block_shape {}; variable2827 void* primitive = mindspore::lite::MindIR_SpaceToBatchND_CreatePrimitive(block_shape, paddings);