Searched refs:out_format_ (Results 1 – 14 of 14) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/op/ |
| D | reduce_tensorrt.cc | 59 out_format_ = tensorrt_in_tensors_[0].format_; in AddInnerOp()
61 MS_LOG(DEBUG) << "origin input " << GetTensorFormat(shuffler_input, out_format_); in AddInnerOp()
73 out_format_ = Format::NHWC; in AddInnerOp()
96 MS_LOG(DEBUG) << "after transpose and expand dims " << GetTensorFormat(reduce_input, out_format_); in AddInnerOp()
124 this->AddInnerOutTensors(ITensorHelper{out_tensor, out_format_}); in AddInnerOp()
125 MS_LOG(DEBUG) << "output " << GetTensorFormat(out_tensor, out_format_); in AddInnerOp()
|
| D | shuffle_tensorrt.cc | 87 out_format_ = Format::NHWC; in AddInnerOp()
95 out_format_ = Format::NCHW; in AddInnerOp()
101 out_format_ = tensorrt_in_tensors_[0].format_; in AddInnerOp()
103 MS_LOG(DEBUG) << "after transpose " << GetTensorFormat(shuffler_input, out_format_); in AddInnerOp()
149 MS_LOG(DEBUG) << "output " << GetTensorFormat(out_tensor, out_format_); in AddInnerOp()
150 this->AddInnerOutTensors(ITensorHelper{out_tensor, out_format_}); in AddInnerOp()
236 out_format_ = Format::NCHW; in AddTransposeOp()
238 out_format_ = Format::NHWC; in AddTransposeOp()
|
| D | scale_tensorrt.cc | 64 out_format_ = tensorrt_in_tensors_[0].format_; in AddInnerOp()
65 …DEBUG) << "before transpose " << GetTensorFormat(tensorrt_in_tensors_[0].trt_tensor_, out_format_); in AddInnerOp()
73 MS_LOG(DEBUG) << "after transpose " << GetTensorFormat(scale_in_tensor, out_format_); in AddInnerOp()
132 this->AddInnerOutTensors(ITensorHelper{op_out_tensor, out_format_}); in AddInnerOp()
133 MS_LOG(DEBUG) << "output " << GetTensorFormat(op_out_tensor, out_format_); in AddInnerOp()
157 out_format_ = Format::NCHW; in PreProcessInputTensor()
168 out_format_ = Format::NHWC; in PreProcessInputTensor()
|
| D | shuffle_tensorrt.h | 45 Format out_format_ = Format::NHWC; variable
|
| D | scale_tensorrt.h | 47 Format out_format_; variable
|
| D | reduce_tensorrt.h | 48 Format out_format_; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | conv_parameter.h | 57 int out_format_; member
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | conv_winograd_fp32.c | 89 if (conv_param->out_format_ != NNACL_NC4HW4) { // nc4hw4 in ConvWinogardFp32()
|
| D | conv_common_fp32.c | 380 int write_mode = conv_param->out_format_; in ConvSWFp32()
|
| D | conv_depthwise_fp32.c | 126 if (conv_param->out_format_ == NNACL_NC4HW4) { in InitSlidingParam()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | convolution_winograd_fp32.cc | 139 conv_param_->out_format_ = out_tensors_[0]->format(); in ReSize()
|
| D | convolution_slidewindow_fp32.cc | 61 conv_param_->out_format_ = out_tensors_[0]->format(); in Init()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | convolution_winograd_fp16.cc | 205 conv_param_->out_format_ = out_tensors_[0]->format(); in ReSize()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | conv_fp16.c | 247 if (conv_param->out_format_ != NNACL_NC4HW4) { // nc4hw4 in ConvWinogardFp16()
|