| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/nnacl/infer/ |
| D | reshape_infer_test.cc | 61 std::vector<int32_t> shape_tensor = {6}; in TEST_F() local
93 std::vector<int8_t> shape_tensor = {6}; in TEST_F() local
125 std::vector<uint32_t> shape_tensor = {6}; in TEST_F() local
156 std::vector<float> shape_tensor = {3.0, 4.0}; in TEST_F() local
188 std::vector<int64_t> shape_tensor = {3, 4}; in TEST_F() local
222 std::vector<int64_t> shape_tensor = {3, 6}; in TEST_F() local
256 std::vector<int64_t> shape_tensor = {3, -1}; in TEST_F() local
289 std::vector<int64_t> shape_tensor = {1, 2, 5, 4}; in TEST_F() local
324 std::vector<int64_t> shape_tensor = {8, 5, -1, 1}; in TEST_F() local
|
| D | resize_infer_test.cc | 69 std::vector<int32_t> shape_tensor = {4, 3, 2, 5}; in TEST_F() local
108 std::vector<float> shape_tensor = {4.0, 3.0, 2.0, 5.0}; in TEST_F() local
147 std::vector<int32_t> shape_tensor = {4, 3, 2, 5}; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/ |
| D | fft_shapecopy_proto.cc | 27 Tensor shape_tensor; in IMPLEMT_COMMON_INFERFUNC() local
|
| D | random_ops_proto.cc | 41 Tensor shape_tensor; in IMPLEMT_COMMON_INFERFUNC() local
347 Tensor shape_tensor; in CUST_IMPLEMT_INFERFUNC() local
402 Tensor shape_tensor; in CUST_IMPLEMT_INFERFUNC() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/coreml/op/ |
| D | reshape_coreml.cc | 26 auto shape_tensor = in_tensors_.at(1); in BuildLayer() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | sparse_reorder.cc | 68 Tensor *shape_tensor = ctx.Input(2); in ValidParam() local
160 Tensor *shape_tensor = ctx.Input(2); in Compute() local
|
| D | fft_shapecopy.cc | 72 auto shape_tensor = reinterpret_cast<int64_t *>(ctx.Input(kFftShapeIndex)->GetData()); in FFTShapeCopyCompute() local
|
| D | sparse_addmm.cc | 94 Tensor *shape_tensor = ctx.Input(2); in SparseAddmmCheck() local
282 auto *shape_tensor = ctx.Input(2); in SparseAddmmCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/ |
| D | crop_and_resize_infer.c | 54 const TensorC *shape_tensor = inputs[3]; in CropAndResizeInferShape() local
|
| D | reshape_infer.c | 104 const TensorC *shape_tensor = inputs[1]; in CalShapeByType() local
204 const TensorC *shape_tensor = inputs[1]; in ReshapeInferShape() local
|
| D | broadcast_to_infer.c | 21 int GetShapeByType(const TensorC *shape_tensor, int shape_size, int *dst_shape) { in GetShapeByType()
172 const TensorC *shape_tensor = inputs[1]; in BroadcastToInferShape() local
|
| D | resize_infer.c | 26 const TensorC *shape_tensor = inputs[1]; in HandleTwoInputs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_split_parser.cc | 39 const auto &shape_tensor = tflite_subgraph->tensors.at(tflite_op->inputs.at(SECOND_INPUT)); in Parse() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/ |
| D | unify_format.cc | 233 int SetTensorInfoData(const tensor::Tensor *shape_tensor, tensor::TensorPtr tensor_info) { in SetTensorInfoData()
244 int SetShapeTensorInfo(const tensor::Tensor *shape_tensor, tensor::TensorPtr tensor_info) { in SetShapeTensorInfo()
268 auto shape_tensor = std::dynamic_pointer_cast<tensor::Tensor>(resize_shape_node->default_param()); in ConvertOnnxResizeForConstShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/ |
| D | oneslike_tensorrt.cc | 70 auto shape_tensor = ctx->network()->addShape(*input_trt_tensor)->getOutput(0); in RunAsTrtOps() local
|
| D | shuffle_tensorrt.cc | 276 auto &shape_tensor = in_tensors_[1]; in AddReshapeOp() local
332 auto shape_tensor = input(ctx_, 1).trt_tensor_; in AddBroadcastToOp() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/kernel/ |
| D | reshape.cc | 126 auto shape_tensor = in_tensors_[1]; in PreProcess() local
|
| D | resize.cc | 153 auto shape_tensor = in_tensors_[1]; in PreProcess() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/npu/op/ |
| D | reshape_npu.cc | 27 auto shape_tensor = in_tensors.at(1); in IsSupport() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/ |
| D | uniform_real_kernels.cc | 79 aicpuops::Tensor shape_tensor = node_def_.inputs(0); in ParseKernelParam() local
|
| D | standard_normal_kernels.cc | 79 aicpuops::Tensor shape_tensor = node_def_.inputs(0); in ParseKernelParam() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/ |
| D | avg_pool_grad_for_ge.cc | 53 …auto shape_tensor = std::make_shared<tensor::Tensor>(value_node_data, TypeIdToType(TypeId::kNumber… in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/pass/format_pass/ |
| D | transpose_strategy.cc | 199 const lite::Tensor &shape_tensor) { in Handle2AxisCrop()
227 const lite::Tensor &shape_tensor) { in Handle3AxisCrop()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | adjust_resize_dims_pass.cc | 308 auto shape_tensor = std::dynamic_pointer_cast<tensor::Tensor>(resize_shape_node->default_param()); in CreateResizeCNodeFor3D() local
401 auto shape_tensor = std::dynamic_pointer_cast<tensor::Tensor>(resize_shape_node->default_param()); in GetResizeInputDims() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | fillv2_info.cc | 165 Shape FillV2Info::GetShapeFromTensor(const tensor::TensorPtr &shape_tensor) { in GetShapeFromTensor()
|