| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | batch_to_space_nd_tests.cc | 26 OpParameter *CreateParameter(int block_shape[], int crops[], const std::vector<int> &input_shape, in CreateParameter()
40 int block_shape[] = {2, 2}; in TEST_F() local
84 int block_shape[] = {3, 3}; in TEST_F() local
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | space_to_batch_nd.cc | 38 auto block_shape = GetValue<std::vector<int64_t>>(primitive->GetAttr(kBlockShape)); in InferShape() local
82 void SpaceToBatchND::set_block_shape(std::vector<int64_t> block_shape) { in set_block_shape()
96 void SpaceToBatchND::Init(const std::vector<int64_t> block_shape, const std::vector<std::vector<int… in Init()
|
| D | batch_to_space_nd.cc | 38 auto block_shape = GetValue<std::vector<int64_t>>(primitive->GetAttr(kBlockShape)); in InferShape() local
84 void BatchToSpaceND::set_block_shape(std::vector<int64_t> block_shape) { in set_block_shape()
99 void BatchToSpaceND::Init(const std::vector<int64_t> block_shape, const std::vector<std::vector<int… in Init()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | space_batch_nd_attr_update.cc | 46 auto block_shape = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(node, kAttrBlockShape); in Process() local
70 auto block_shape = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(node, kAttrBlockShape); in Process() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | batch_to_space_infer.c | 30 const int32_t *block_shape = param->block_shape_; in SetOutputShapeFromParam() local
72 int *block_shape = (int *)(inputs[1]->data_); in SetOutputShapeFromInput() local
|
| D | space_to_batch_nd_infer.c | 31 const int *block_shape = param->block_sizes_; in SpaceSetOutputShapeFromParam() local
73 int *block_shape = (int *)(inputs[1]->data_); in SpaceSetOutputShapeFromInput() local
|
| D | space_to_batch_infer.c | 40 int *block_shape = param->block_sizes_; in SpaceToBatchInferShape() local
|
| /third_party/mindspore/mindspore/lite/src/ops/populate/v0/ |
| D | batch_to_space_populate_v0.cc | 31 auto block_shape = attr->blockShape(); in PopulateBatchToSpaceParameter() local
|
| /third_party/mindspore/mindspore/lite/src/ops/populate/ |
| D | batch_to_space_populate.cc | 41 auto block_shape = std::vector<int64_t>(block_size->begin(), block_size->end()); in PopulateBatchToSpaceParameter() local
|
| D | space_to_batch_nd_populate.cc | 39 auto block_shape = value->block_shape(); in PopulateSpaceToBatchNDParameter() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | batch_to_space_fp32.cc | 35 auto block_shape = static_cast<int *>(block_shape_data); in Processinput() local
|
| D | space_to_batch_fp32.cc | 38 auto block_shape = static_cast<int *>(block_shape_data); in ProcessInput() local
|
| /third_party/mindspore/tests/st/networks/models/deeplabv3/src/backbone/ |
| D | resnet_deeplab.py | 186 def __init__(self, block_shape, paddings): argument
197 def __init__(self, block_shape, crops): argument
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | array_ops.py | 5173 def __init__(self, block_shape, paddings): argument
5277 def __init__(self, block_shape, crops): argument
|
| /third_party/mindspore/mindspore/lite/mindir/src/ |
| D | mindir.cc | 696 PrimitivePtr MindIR_BatchToSpaceND_CreatePrimitive(const std::vector<int64_t> &block_shape, in MindIR_BatchToSpaceND_CreatePrimitive()
726 …IR_BatchToSpaceND_SetBlockShape(PrimitivePtr *primitive, const std::vector<int64_t> &block_shape) { in MindIR_BatchToSpaceND_SetBlockShape()
3562 PrimitivePtr MindIR_SpaceToBatchND_CreatePrimitive(const std::vector<int64_t> &block_shape, in MindIR_SpaceToBatchND_CreatePrimitive()
3592 …IR_SpaceToBatchND_SetBlockShape(PrimitivePtr *primitive, const std::vector<int64_t> &block_shape) { in MindIR_SpaceToBatchND_SetBlockShape()
|