| /third_party/mindspore/mindspore/core/ops/ |
| D | space_to_batch_nd.h | 33 class MS_CORE_API SpaceToBatchND : public PrimitiveC {
36 SpaceToBatchND() : PrimitiveC(kNameSpaceToBatchND) {} in SpaceToBatchND() function
38 ~SpaceToBatchND() = default;
39 MS_DECLARE_PARENT(SpaceToBatchND, PrimitiveC);
57 using PrimSpaceToBatchNDPtr = std::shared_ptr<SpaceToBatchND>;
|
| D | space_to_batch_nd.cc | 63 void SpaceToBatchND::set_paddings(std::vector<std::vector<int64_t>> paddings) { in set_paddings()
78 std::vector<std::vector<int64_t>> SpaceToBatchND::get_paddings() const { in get_paddings()
82 void SpaceToBatchND::set_block_shape(std::vector<int64_t> block_shape) { in set_block_shape()
92 std::vector<int64_t> SpaceToBatchND::get_block_shape() const { in get_block_shape()
96 void SpaceToBatchND::Init(const std::vector<int64_t> block_shape, const std::vector<std::vector<int… in Init()
106 REGISTER_PRIMITIVE_C(kNameSpaceToBatchND, SpaceToBatchND);
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | tf_space_to_batch_nd_parser.cc | 29 auto prim = std::make_unique<ops::SpaceToBatchND>(); in Parse()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_space_to_batch_nd_parser.cc | 33 auto prim = std::make_unique<ops::SpaceToBatchND>(); in Parse()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | space_to_batch_nd_infer.c | 142 REG_INFER(SpaceToBatchND, PrimType_SpaceToBatchND, SpaceToBatchNdInferShape)
|
| /third_party/mindspore/mindspore/lite/schema/ |
| D | model_v0.fbs | 201 SpaceToBatchND,
|
| D | ops.fbs | 161 SpaceToBatchND,
928 table SpaceToBatchND {
|
| D | ops_v0.fbs | 1008 table SpaceToBatchND {
|
| D | model_generated.h | 887 const mindspore::schema::SpaceToBatchND *value_as_SpaceToBatchND() const { in value_as_SpaceToBatchND()
888 …:PrimitiveType_SpaceToBatchND ? static_cast<const mindspore::schema::SpaceToBatchND *>(value()) : … in value_as_SpaceToBatchND()
1636 …emplate<> inline const mindspore::schema::SpaceToBatchND *Primitive::value_as<mindspore::schema::S…
|
| D | model_v0_generated.h | 1281 template<> struct PrimitiveTypeTraits<mindspore::schema::v0::SpaceToBatchND> {
2325 const mindspore::schema::v0::SpaceToBatchND *value_as_SpaceToBatchND() const {
2326 …mitiveType_SpaceToBatchND ? static_cast<const mindspore::schema::v0::SpaceToBatchND *>(value()) : …
3089 …e<> inline const mindspore::schema::v0::SpaceToBatchND *Primitive::value_as<mindspore::schema::v0:…
4327 auto ptr = reinterpret_cast<const mindspore::schema::v0::SpaceToBatchND *>(obj);
|
| D | ops_generated.h | 437 struct SpaceToBatchND;
1805 template<> struct PrimitiveTypeTraits<mindspore::schema::SpaceToBatchND> {
9643 struct SpaceToBatchND FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table {
9666 typedef SpaceToBatchND Table;
9670 fbb_.AddOffset(SpaceToBatchND::VT_BLOCK_SHAPE, block_shape);
9673 fbb_.AddOffset(SpaceToBatchND::VT_PADDINGS, paddings);
9679 flatbuffers::Offset<SpaceToBatchND> Finish() {
9681 auto o = flatbuffers::Offset<SpaceToBatchND>(end);
9686 inline flatbuffers::Offset<SpaceToBatchND> CreateSpaceToBatchND(
9696 inline flatbuffers::Offset<SpaceToBatchND> CreateSpaceToBatchNDDirect(
[all …]
|
| D | ops_v0_generated.h | 481 struct SpaceToBatchND;
10355 struct SpaceToBatchND FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table {
10378 typedef SpaceToBatchND Table;
10382 fbb_.AddOffset(SpaceToBatchND::VT_BLOCKSHAPE, blockShape); in add_blockShape()
10385 fbb_.AddOffset(SpaceToBatchND::VT_PADDINGS, paddings); in add_paddings()
10391 flatbuffers::Offset<SpaceToBatchND> Finish() { in Finish()
10393 auto o = flatbuffers::Offset<SpaceToBatchND>(end); in Finish()
10398 inline flatbuffers::Offset<SpaceToBatchND> CreateSpaceToBatchND(
10408 inline flatbuffers::Offset<SpaceToBatchND> CreateSpaceToBatchNDDirect(
|
| /third_party/mindspore/mindspore/lite/src/ops/ |
| D | ops_func_declare.h | 408 FUNC_MSOP2SCHEMAOP_DECLARE(SpaceToBatchND)
|
| D | ops_def.cc | 160 OP_TYPE(SpaceToBatchND)
928 OP_SCHEMA_DEF(SpaceToBatchND)
931 OP_SCHEMA_DEF_END(SpaceToBatchND)
|
| D | ops_utils.cc | 629 auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::SpaceToBatchND>>(node); in SpaceToBatchNDPrimitiveCreator()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 35 … BatchToSpace, SpaceToBatchND, BatchToSpaceND, BroadcastTo, InplaceUpdate, ReverseSequence,
|
| D | array_ops.py | 5114 class SpaceToBatchND(PrimitiveWithInfer): class
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_array_ops.py | 312 self.space_to_batch_nd = P.SpaceToBatchND(block_shape, paddings)
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_array_ops.py | 1002 @bprop_getters.register(P.SpaceToBatchND)
1017 batch_to_space_nd_grad = P.SpaceToBatchND(self.block_shape, self.crops)
|
| /third_party/mindspore/mindspore/lite/schema/inner/ |
| D | model_generated.h | 1036 const mindspore::schema::SpaceToBatchND *value_as_SpaceToBatchND() const { in value_as_SpaceToBatchND()
1037 …:PrimitiveType_SpaceToBatchND ? static_cast<const mindspore::schema::SpaceToBatchND *>(value()) : … in value_as_SpaceToBatchND()
1791 …emplate<> inline const mindspore::schema::SpaceToBatchND *Primitive::value_as<mindspore::schema::S…
|
| D | ops_v0_generated.h | 637 struct SpaceToBatchND;
14558 typedef SpaceToBatchND TableType;
14563 struct SpaceToBatchND FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table {
14595 …static flatbuffers::Offset<SpaceToBatchND> Pack(flatbuffers::FlatBufferBuilder &_fbb, const SpaceT…
14599 typedef SpaceToBatchND Table;
14603 fbb_.AddOffset(SpaceToBatchND::VT_BLOCKSHAPE, blockShape); in add_blockShape()
14606 fbb_.AddOffset(SpaceToBatchND::VT_PADDINGS, paddings); in add_paddings()
14612 flatbuffers::Offset<SpaceToBatchND> Finish() { in Finish()
14614 auto o = flatbuffers::Offset<SpaceToBatchND>(end); in Finish()
14619 inline flatbuffers::Offset<SpaceToBatchND> CreateSpaceToBatchND(
[all …]
|
| D | ops_generated.h | 578 struct SpaceToBatchND;
2405 template<> struct PrimitiveTypeTraits<mindspore::schema::SpaceToBatchND> {
14839 typedef SpaceToBatchND TableType;
14844 struct SpaceToBatchND FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table {
14876 …static flatbuffers::Offset<SpaceToBatchND> Pack(flatbuffers::FlatBufferBuilder &_fbb, const SpaceT…
14880 typedef SpaceToBatchND Table;
14884 fbb_.AddOffset(SpaceToBatchND::VT_BLOCK_SHAPE, block_shape);
14887 fbb_.AddOffset(SpaceToBatchND::VT_PADDINGS, paddings);
14893 flatbuffers::Offset<SpaceToBatchND> Finish() {
14895 auto o = flatbuffers::Offset<SpaceToBatchND>(end);
[all …]
|
| D | model_v0_generated.h | 1299 template<> struct PrimitiveTypeTraits<mindspore::schema::v0::SpaceToBatchND> {
4132 const mindspore::schema::v0::SpaceToBatchND *value_as_SpaceToBatchND() const {
4133 …mitiveType_SpaceToBatchND ? static_cast<const mindspore::schema::v0::SpaceToBatchND *>(value()) : …
4902 …e<> inline const mindspore::schema::v0::SpaceToBatchND *Primitive::value_as<mindspore::schema::v0:…
6531 auto ptr = reinterpret_cast<const mindspore::schema::v0::SpaceToBatchND *>(obj);
7365 auto ptr = reinterpret_cast<const mindspore::schema::v0::SpaceToBatchND *>(obj);
|
| /third_party/mindspore/mindspore/lite/mindir/src/ |
| D | mindir_nnrt_lite_graph_to_model.cc | 1068 SpaceToBatchND space_to_batch_n_d{}; in ConvertSpaceToBatchND()
|
| /third_party/mindspore/ |
| D | RELEASE.md | 1829 ##### `ops.SpaceToBatch`, `ops.BatchToSpace` are deprecated in favor of `ops.SpaceToBatchND`, `ops.…
1831 The `ops.SpaceToBatchND`, `ops.BatchToSpaceND` are more general and have same behavior as `ops.Spac…
|