Searched refs:kBlockShape (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | batch_to_space_nd.cc | 38 auto block_shape = GetValue<std::vector<int64_t>>(primitive->GetAttr(kBlockShape)); in InferShape()
86 …(void)CheckAndConvertUtils::CheckInteger(kBlockShape, SizeToLong(block_shape.size()), kEqual, bloc… in set_block_shape()
89 …(void)CheckAndConvertUtils::CheckInteger(kBlockShape, block_shape[i], kGreaterEqual, 1, this->name… in set_block_shape()
91 (void)this->AddAttr(kBlockShape, MakeValue(block_shape)); in set_block_shape()
95 auto value_ptr = GetAttr(kBlockShape); in get_block_shape()
|
| D | space_to_batch_nd.cc | 38 auto block_shape = GetValue<std::vector<int64_t>>(primitive->GetAttr(kBlockShape)); in InferShape()
84 …(void)CheckAndConvertUtils::CheckInteger(kBlockShape, SizeToLong(block_shape.size()), kEqual, bloc… in set_block_shape()
87 …(void)CheckAndConvertUtils::CheckInteger(kBlockShape, block_shape[i], kGreaterEqual, 1LL, this->na… in set_block_shape()
89 (void)this->AddAttr(kBlockShape, MakeValue(block_shape)); in set_block_shape()
93 return GetValue<std::vector<int64_t>>(GetAttr(kBlockShape)); in get_block_shape()
|
| D | op_utils.h | 50 constexpr auto kBlockShape = "block_shape"; variable
|