Searched refs:nhwc_shape (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/delegate/npu/pass/ |
| D | npu_transform_pass.cc | 48 auto nhwc_shape = op->inputs()[0].Shape(); in InsertPreNodes() local
49 …std::vector<int64_t> nchw_shape = {nhwc_shape[NHWC_N], nhwc_shape[NHWC_C], nhwc_shape[NHWC_H], nhw… in InsertPreNodes()
104 auto nhwc_shape = op->outputs()[0].Shape(); in InsertPostNodes() local
105 …std::vector<int64_t> nchw_shape = {nhwc_shape[NHWC_N], nhwc_shape[NHWC_C], nhwc_shape[NHWC_H], nhw… in InsertPostNodes()
|
| D | npu_insert_transform_pass.cc | 133 auto nhwc_shape = in_tensor.Shape(); in InsertNode() local
134 if (nhwc_shape.size() == 0) { in InsertNode()
136 } else if (nhwc_shape.size() < 4) { in InsertNode()
140 std::vector<int64_t> nchw_shape = {nhwc_shape[0], nhwc_shape[3], nhwc_shape[1], nhwc_shape[2]}; in InsertNode()
155 …mindspore::MSTensor::CreateTensor(nc2nh_name + "/output0", in_tensor.DataType(), nhwc_shape, nullp… in InsertNode()
|
| D | npu_fusion_pass.cc | 201 auto nhwc_shape = tensor.Shape(); in UpdatePostTensors() local
202 if (nhwc_shape.size() < kNumDims) { in UpdatePostTensors()
206 tensor.SetShape({nhwc_shape[NHWC_N], nhwc_shape[NHWC_C], nhwc_shape[NHWC_H], nhwc_shape[NHWC_W]}); in UpdatePostTensors()
|
| /third_party/mindspore/mindspore/lite/src/runtime/ |
| D | runtime_pass.cc | 56 … std::vector<int> nhwc_shape = {nwhc_tensor->Batch(), nwhc_tensor->Height(), nwhc_tensor->Width(), in Nc4hw4PassReplace() local
59 nwhc_tensor->set_shape(nhwc_shape); in Nc4hw4PassReplace()
|