Home
last modified time | relevance | path

Searched defs:block_shape (Results 1 – 25 of 29) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dbatch_to_space_nd_tests.cc26 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-src/source/tests/st/ops/cpu/
Dtest_batchtospacend_op.py30 def __init__(self, nptype, block_shape=2, input_shape=(4, 1, 1, 1)): argument
43 def batch_to_space_nd_test_case(nptype, block_shape=2, input_shape=(4, 1, 1, 1)): argument
99 def __init__(self, block_shape, crops, is_dynamic_rank): argument
150 def __init__(self, block_shape, crops): argument
/third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/
Dtest_batchtospacend_op.py27 def __init__(self, nptype, block_shape=2, input_shape=(4, 1, 1, 1)): argument
40 def batch_to_space_nd_test_case(nptype, block_shape=2, input_shape=(4, 1, 1, 1)): argument
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dbatch_to_space_nd.cc71 auto block_shape = GetValue<std::vector<int64_t>>(primitive->GetAttr(kBlockShape)); in BatchToSpaceNDInferShape() local
128 void BatchToSpaceND::set_block_shape(std::vector<int64_t> block_shape) { in set_block_shape()
143 void BatchToSpaceND::Init(const std::vector<int64_t> block_shape, const std::vector<std::vector<int… in Init()
Dspace_to_batch_nd.cc174 void SpaceToBatchND::set_block_shape(std::vector<int64_t> block_shape) { in set_block_shape()
196 void SpaceToBatchND::Init(const std::vector<int64_t> block_shape, const std::vector<std::vector<int… in Init()
Dbatch_to_space_nd_v2.cc62 …auto block_shape = CheckAndConvertUtils::CheckTensorIntValue(kBlockShape, input_args[1]->GetValue(… in BatchToSpaceNDV2InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/
Dbatch_to_space_infer.c31 const int32_t *block_shape = param->block_shape_; in SetOutputShapeFromParam() local
73 int *block_shape = (int *)(inputs[1]->data_); in SetOutputShapeFromInput() local
Dspace_to_batch_nd_infer.c32 const int *block_shape = param->block_sizes_; in SpaceSetOutputShapeFromParam() local
74 int *block_shape = (int *)(inputs[1]->data_); in SpaceSetOutputShapeFromInput() local
Dspace_to_batch_infer.c40 int *block_shape = param->block_sizes_; in SpaceToBatchInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dbatch_to_space_nd_impl.cu26 …BatchToSpaceND(const T *__restrict__ input, const int64_t *crops_start, const int64_t *block_shape, in BatchToSpaceND()
62 …daError_t CalBatchToSpaceND(const T *input, const int64_t *crops_start, const int64_t *block_shape, in CalBatchToSpaceND()
Dspace_to_batch_nd_impl.cu26 …ceToBatchND(const T *__restrict__ input, const int64_t *paddings_start, const int64_t *block_shape, in SpaceToBatchND()
62 …rror_t CalSpaceToBatchND(const T *input, const int64_t *paddings_start, const int64_t *block_shape, in CalSpaceToBatchND()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/
Dspace_batch_nd_attr_update.cc42 auto block_shape = common::AnfAlgo::GetNodeAttr<std::vector<int64_t>>(node, kAttrBlockShape); in BuildSpace() local
57 auto block_shape = common::AnfAlgo::GetNodeAttr<std::vector<int64_t>>(node, kAttrBlockShape); in BuildBatch() local
/third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/
Dtest_batchtospacend_op.py53 def construct(self, x, block_shape, crops, indices): argument
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/common/ops/populate/
Dbatch_to_space_populate.cc41 auto block_shape = std::vector<int64_t>(block_size->begin(), block_size->end()); in PopulateBatchToSpaceParameter() local
Dspace_to_batch_nd_populate.cc39 auto block_shape = value->block_shape(); in PopulateSpaceToBatchNDParameter() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/common/ops/operator_populate/
Dbatch_to_space_populate.cc46 auto block_shape = std::vector<int64_t>(block_size.begin(), block_size.end()); in PopulateBatchToSpaceOpParameter() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/
Dbatch_to_space.c25 TensorC *block_shape = batch_to_space->base_.in_[SECOND_INPUT]; in BatchToSpaceProcessInput() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/
Dspace_to_batch_fp32.cc30 auto block_shape = static_cast<int *>(block_shape_data); in ProcessInput() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/int8/
Dbatch_to_space_int8.cc48 auto block_shape = static_cast<int *>(block_shape_data); in Processinput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_class/
Dspace_to_batch_nd_helper.h32 std::vector<int64_t> block_shape; variable
Dbatch_to_space_nd_helper.h32 std::vector<int64_t> block_shape; variable
/third_party/mindspore/mindspore-src/source/tests/st/networks/models/deeplabv3/src/backbone/
Dresnet_deeplab.py186 def __init__(self, block_shape, paddings): argument
197 def __init__(self, block_shape, crops): argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/
Ddeprecated.py274 def __init__(self, block_shape, crops): argument
/third_party/mesa3d/src/intel/vulkan/
Danv_sparse.c249 VkExtent3D block_shape = { .width = 0, .height = 0, .depth = 0 }; in anv_sparse_get_standard_image_block_shape() local
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
Darray_ops.py3744 def __init__(self, block_shape, paddings): argument

12