BatchToSpaceND (reference) in projects: third_party

Project(s)

Full Search
Definition
Symbol
File Path
History
Type

Searched refs:BatchToSpaceND (Results 1 – 23 of 23) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
D	batch_to_space_nd.h	`33 class MS_CORE_API BatchToSpaceND : public PrimitiveC { 36 BatchToSpaceND() : PrimitiveC(kNameBatchToSpaceND) {} in BatchToSpaceND() function 38 ~BatchToSpaceND() = default; 39 MS_DECLARE_PARENT(BatchToSpaceND, PrimitiveC); 57 using PrimBatchToSpaceNDPtr = std::shared_ptr<BatchToSpaceND>;`
D	batch_to_space_nd.cc	65 void BatchToSpaceND::set_crops(std::vector<std::vector<int64_t>> crops) { in set_crops() 80 std::vector<std::vector<int64_t>> BatchToSpaceND::get_crops() const { in get_crops() 84 void BatchToSpaceND::set_block_shape(std::vector<int64_t> block_shape) { in set_block_shape() 94 std::vector<int64_t> BatchToSpaceND::get_block_shape() const { in get_block_shape() 99 void BatchToSpaceND::Init(const std::vector<int64_t> block_shape, const std::vector<std::vector<int… in Init() 107 REGISTER_PRIMITIVE_C(kNameBatchToSpaceND, BatchToSpaceND);
/third_party/mindspore/mindspore/lite/schema/
D	model_v0.fbs	`211 BatchToSpaceND,`
D	ops.fbs	`42 BatchToSpaceND, 340 table BatchToSpaceND {`
D	ops_v0.fbs	`850 table BatchToSpaceND {`
D	model_generated.h	`530 const mindspore::schema::BatchToSpaceND value_as_BatchToSpaceND() const { in value_as_BatchToSpaceND() 531 …:PrimitiveType_BatchToSpaceND ? static_cast<const mindspore::schema::BatchToSpaceND >(value()) : … in value_as_BatchToSpaceND() 1160 …emplate<> inline const mindspore::schema::BatchToSpaceND *Primitive::value_as<mindspore::schema::B…`
D	model_v0_generated.h	`1321 template<> struct PrimitiveTypeTraits<mindspore::schema::v0::BatchToSpaceND> { 2355 const mindspore::schema::v0::BatchToSpaceND value_as_BatchToSpaceND() const { 2356 …mitiveType_BatchToSpaceND ? static_cast<const mindspore::schema::v0::BatchToSpaceND >(value()) : … 3129 …e<> inline const mindspore::schema::v0::BatchToSpaceND Primitive::value_as<mindspore::schema::v0:… 4367 auto ptr = reinterpret_cast<const mindspore::schema::v0::BatchToSpaceND >(obj);`
D	ops_generated.h	77 struct BatchToSpaceND; 1329 template<> struct PrimitiveTypeTraits<mindspore::schema::BatchToSpaceND> { 3322 struct BatchToSpaceND FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table { 3345 typedef BatchToSpaceND Table; 3349 fbb_.AddOffset(BatchToSpaceND::VT_BLOCK_SHAPE, block_shape); 3352 fbb_.AddOffset(BatchToSpaceND::VT_CROPS, crops); 3358 flatbuffers::Offset<BatchToSpaceND> Finish() { 3360 auto o = flatbuffers::Offset<BatchToSpaceND>(end); 3365 inline flatbuffers::Offset<BatchToSpaceND> CreateBatchToSpaceND( 3375 inline flatbuffers::Offset<BatchToSpaceND> CreateBatchToSpaceNDDirect( [all …]
D	ops_v0_generated.h	364 struct BatchToSpaceND; 8700 struct BatchToSpaceND FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table { 8723 typedef BatchToSpaceND Table; 8727 fbb_.AddOffset(BatchToSpaceND::VT_BLOCKSHAPE, blockShape); in add_blockShape() 8730 fbb_.AddOffset(BatchToSpaceND::VT_CROPS, crops); in add_crops() 8736 flatbuffers::Offset<BatchToSpaceND> Finish() { in Finish() 8738 auto o = flatbuffers::Offset<BatchToSpaceND>(end); in Finish() 8743 inline flatbuffers::Offset<BatchToSpaceND> CreateBatchToSpaceND( 8753 inline flatbuffers::Offset<BatchToSpaceND> CreateBatchToSpaceNDDirect(
/third_party/mindspore/mindspore/lite/src/ops/
D	ops_func_declare.h	`281 FUNC_MSOP2SCHEMAOP_DECLARE(BatchToSpaceND)`
D	ops_def.cc	`41 OP_TYPE(BatchToSpaceND) 340 OP_SCHEMA_DEF(BatchToSpaceND) 343 OP_SCHEMA_DEF_END(BatchToSpaceND)`
D	ops_utils.cc	`133 auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::BatchToSpaceND>>(node); in BatchToSpaceNDPrimitiveCreator()`
/third_party/mindspore/mindspore/ops/operations/
D	__init__.py	`35 … BatchToSpace, SpaceToBatchND, BatchToSpaceND, BroadcastTo, InplaceUpdate, ReverseSequence,`
D	array_ops.py	`5222 class BatchToSpaceND(PrimitiveWithInfer): class`
/third_party/mindspore/tests/ut/python/ops/
D	test_array_ops.py	`301 self.batch_to_space_nd = P.BatchToSpaceND(block_shape, crops)`
/third_party/mindspore/mindspore/ops/_grad/
D	grad_array_ops.py	`1005 space_to_batch_nd_grad = P.BatchToSpaceND(self.block_shape, self.paddings) 1014 @bprop_getters.register(P.BatchToSpaceND)`
/third_party/mindspore/mindspore/lite/schema/inner/
D	model_generated.h	`679 const mindspore::schema::BatchToSpaceND value_as_BatchToSpaceND() const { in value_as_BatchToSpaceND() 680 …:PrimitiveType_BatchToSpaceND ? static_cast<const mindspore::schema::BatchToSpaceND >(value()) : … in value_as_BatchToSpaceND() 1315 …emplate<> inline const mindspore::schema::BatchToSpaceND *Primitive::value_as<mindspore::schema::B…`
D	ops_v0_generated.h	481 struct BatchToSpaceND; 12240 typedef BatchToSpaceND TableType; 12245 struct BatchToSpaceND FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table { 12277 …static flatbuffers::Offset<BatchToSpaceND> Pack(flatbuffers::FlatBufferBuilder &_fbb, const BatchT… 12281 typedef BatchToSpaceND Table; 12285 fbb_.AddOffset(BatchToSpaceND::VT_BLOCKSHAPE, blockShape); in add_blockShape() 12288 fbb_.AddOffset(BatchToSpaceND::VT_CROPS, crops); in add_crops() 12294 flatbuffers::Offset<BatchToSpaceND> Finish() { in Finish() 12296 auto o = flatbuffers::Offset<BatchToSpaceND>(end); in Finish() 12301 inline flatbuffers::Offset<BatchToSpaceND> CreateBatchToSpaceND( [all …]
D	ops_generated.h	98 struct BatchToSpaceND; 1929 template<> struct PrimitiveTypeTraits<mindspore::schema::BatchToSpaceND> { 6046 typedef BatchToSpaceND TableType; 6051 struct BatchToSpaceND FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table { 6083 …static flatbuffers::Offset<BatchToSpaceND> Pack(flatbuffers::FlatBufferBuilder &_fbb, const BatchT… 6087 typedef BatchToSpaceND Table; 6091 fbb_.AddOffset(BatchToSpaceND::VT_BLOCK_SHAPE, block_shape); 6094 fbb_.AddOffset(BatchToSpaceND::VT_CROPS, crops); 6100 flatbuffers::Offset<BatchToSpaceND> Finish() { 6102 auto o = flatbuffers::Offset<BatchToSpaceND>(end); [all …]
D	model_v0_generated.h	1339 template<> struct PrimitiveTypeTraits<mindspore::schema::v0::BatchToSpaceND> { 4162 const mindspore::schema::v0::BatchToSpaceND value_as_BatchToSpaceND() const { 4163 …mitiveType_BatchToSpaceND ? static_cast<const mindspore::schema::v0::BatchToSpaceND >(value()) : … 4942 …e<> inline const mindspore::schema::v0::BatchToSpaceND Primitive::value_as<mindspore::schema::v0:… 6571 auto ptr = reinterpret_cast<const mindspore::schema::v0::BatchToSpaceND >(obj); 7405 auto ptr = reinterpret_cast<const mindspore::schema::v0::BatchToSpaceND *>(obj);
/third_party/mindspore/mindspore/lite/mindir/src/
D	mindir_nnrt_lite_graph_to_model.cc	`294 BatchToSpaceND batch_to_space_n_d{}; in ConvertBatchToSpaceND()`
/third_party/mindspore/
D	RELEASE.md	1829 ….BatchToSpace` are deprecated in favor of `ops.SpaceToBatchND`, `ops.BatchToSpaceND`([!11527](http… 1831 The `ops.SpaceToBatchND`, `ops.BatchToSpaceND` are more general and have same behavior as `ops.Spac… 2764 - Fix BatchToSpaceND([!2741](https://gitee.com/mindspore/mindspore/pulls/2741))
/third_party/mindspore/config/
D	op_info.config	`427 {"op_name": "BatchToSpaceND", "inputs": [{"index": 0, "name": "x", "need_compile": false, "param_ty…`