| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/parallel/tensor_layout/ |
| D | reshape_layout_transfer_test.cc | 49 Arrangement in_tensor_shape; in InferUnifiedLayout() local
89 const TensorShape &in_tensor_shape, const TensorMap &out_tensor_map, in InferUnifiedLayoutCheck()
113 const TensorShape &in_tensor_shape, const TensorMap &out_tensor_map, in ValidUnifiedLayoutCheck()
133 TensorShape in_tensor_shape = {512, 1024}; in TEST_F() local
148 TensorShape in_tensor_shape = {512, 1024}; in TEST_F() local
163 TensorShape in_tensor_shape = {512, 1024}; in TEST_F() local
172 TensorShape in_tensor_shape = {512, 1024}; in TEST_F() local
181 TensorShape in_tensor_shape = {4, 2}; in TEST_F() local
190 TensorShape in_tensor_shape = {8, 2, 2}; in TEST_F() local
199 TensorShape in_tensor_shape = {2, 2}; in TEST_F() local
[all …]
|
| D | tensor_redistribution_test.cc | 63 Arrangement in_tensor_shape; in TEST_F() local
116 Arrangement in_tensor_shape; in TEST_F() local
168 Arrangement in_tensor_shape; in TEST_F() local
|
| D | redistribution_operator_infer_test.cc | 81 Arrangement in_tensor_shape; in InferOperatorCheckAll() local
|
| D | util_layout_gen_test.cc | 266 … const TensorMap &in_tensor_map, const TensorShape &in_tensor_shape, in LayoutTransferValidLayoutChangeCheck()
318 … const TensorShape &in_tensor_shape, const DeviceArrangement &out_device_arrangement, in ValidLayoutChangeCheck()
|
| D | tensor_layout_test.cc | 219 TensorShape in_tensor_shape = {512, 1024}; in TEST_F() local
237 TensorShape in_tensor_shape = {512, 1024}; in TEST_F() local
|
| D | redistribution_layout_transfer_test.cc | 254 TensorShape in_tensor_shape = std::get<2>(layout_list[in]); in ValidRedistributionLayoutCheckAll() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/data/ |
| D | unique_op.cc | 30 auto in_tensor_shape = in_tensor->shape(); in Compute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/wrapper/fp32/ |
| D | split_fp32_wrapper.h | 25 int *in_tensor_shape; member
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/ |
| D | glu_tensorrt.cc | 39 auto in_tensor_shape = ctx->network()->addShape(*input(ctx, 0).trt_tensor_)->getOutput(0); in AddInnerOp() local
|
| D | slicefusion_tensorrt.cc | 26 auto in_tensor_shape = ctx->network()->addShape(*input)->getOutput(0); in GetDynamicSliceSize() local
|
| D | split_tensorrt.cc | 35 auto in_tensor_shape = ctx->network()->addShape(*input)->getOutput(0); in GetDynamicSliceSize() local
|
| D | strideslice_tensorrt.cc | 53 auto in_tensor_shape = ctx->network()->addShape(*input)->getOutput(0); in GetDynamicAxisSliceSize() local
129 auto in_tensor_shape = ctx->network()->addShape(*input)->getOutput(0); in GetDynamicSliceSize() local
|
| D | resize_tensorrt.cc | 113 auto in_tensor_shape = ctx->network()->addShape(*resize_in_tensor)->getOutput(0); in SetOutputDims() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/pass/format_pass/ |
| D | pass_utils.cc | 102 auto in_tensor_shape = in_tensor->shape(); in InsertPreTranspose() local
|