Home
last modified time | relevance | path

Searched defs:tensor_shape (Results 1 – 25 of 104) sorted by relevance

12345

/third_party/mindspore/mindspore-src/source/tests/ut/cpp/parallel/tensor_layout/
Dtensor_layout_test.cc47 Arrangement tensor_shape; in ReshapeExpandDeviceArrangementTestFunction() local
76 TensorShape tensor_shape = {512, 1024}; in TEST_F() local
97 TensorShape tensor_shape = {512, 1024}; in TEST_F() local
117 TensorShape tensor_shape = {512, 1024}; in TEST_F() local
138 TensorShape tensor_shape = {512, 1024}; in TEST_F() local
149 TensorShape tensor_shape = {128, 4, 1024}; in TEST_F() local
167 Arrangement tensor_shape; in ExpandTensorShapeTestFunction() local
197 TensorShape tensor_shape = {512, 1024}; in TEST_F() local
208 TensorShape tensor_shape = {128, 4096}; in TEST_F() local
224 Arrangement tensor_shape; in TEST_F() local
[all …]
Dredistribution_layout_transfer_test.cc49 Arrangement tensor_shape; in RedistributionLayoutTransferTestFunction() local
90 … const TensorShape &tensor_shape, const DeviceArrangement &out_device_arrangement, in RedistributionLayoutCheck()
135 TensorShape tensor_shape = {512, 1024}; in TEST_F() local
170 TensorShape tensor_shape = {512, 1024}; in TEST_F() local
185 TensorShape tensor_shape = {512, 1024}; in TEST_F() local
200 TensorShape tensor_shape = {512, 1024}; in TEST_F() local
215 TensorShape tensor_shape = {512, 1024}; in TEST_F() local
228 … const TensorShape &tensor_shape, const DeviceArrangement &out_device_arrangement, in ValidRedistributionLayoutCheck()
Dtensor_redistribution_test.cc55 TensorShape tensor_shape = {512, 1024}; in TEST_F() local
108 TensorShape tensor_shape = {512, 1024}; in TEST_F() local
160 TensorShape tensor_shape = {128, 64, 1, 1}; in TEST_F() local
Dutil_layout_gen_test.cc106 …enerateValidTensorMap(const DeviceArrangement &device_arrangement, const TensorShape &tensor_shape, in GenerateValidTensorMap()
177 const TensorShape &tensor_shape) { in CheckLayoutValid()
199 const TensorShape &tensor_shape) { in ShapeIsDividedByDevice()
237 const TensorShape &tensor_shape, in ComputeAccumDeviceTOAccumShapeMap()
Dconstruct_operator_test.cc68 Shape tensor_shape = {512, 1024}; in SetUp() local
91 Arrangement tensor_shape; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp16/
Darithmetic_dynamic_fp16_coder.cc122 auto tensor_shape = filter_tensor_->shape(); in InitDynamicParams() local
167 auto tensor_shape = filter_tensor_->shape(); in ResetStatus() local
232 auto tensor_shape = filter_tensor_->shape(); in DoBroadcast() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dfft_with_size.cc163 Eigen::DSizes<Eigen::DenseIndex, signal_ndim + 1> tensor_shape; in GetFlatShape() local
196 uint32_t compute(CpuKernelContext &ctx, bool onesided, in compute()
211 uint32_t compute( // irfft 1d-3d in compute()
286 uint32_t compute( // rfft 1d-3d in compute()
318 uint32_t compute( // fft and ifft 1d-3d in compute()
344 …Eigen::DSizes<Eigen::DenseIndex, signal_ndim + 1> tensor_shape = GetFlatShape<signal_ndim>(x_shape… in FFTWithSizeCompute() local
Dfftnbase.cc93 std::vector<int64_t> tensor_shape, std::vector<int64_t> dim) { in EigenFFTNBase()
123 std::vector<int64_t> tensor_shape = ctx.Input(kIndex0)->GetTensorShape()->GetDimSizes(); in FFTNGetInputs() local
170 std::vector<int64_t> tensor_shape = ctx.Input(kIndex0)->GetTensorShape()->GetDimSizes(); in FFTNBaseCompute() local
Ddct.cc153 Eigen::array<Eigen::DenseIndex, x_rank> tensor_shape; in GetTensorShape() local
270 Eigen::array<Eigen::DenseIndex, x_rank> tensor_shape; in ConstructW4Tensor() local
296 Eigen::array<Eigen::DenseIndex, x_rank> tensor_shape; in ConstructW4TensorForComplex() local
320 Eigen::array<Eigen::DenseIndex, x_rank> tensor_shape; in OrthoNormalize() local
382 …nsor<T, x_rank, Eigen::RowMajor> in_tensor, Eigen::array<Eigen::DenseIndex, x_rank> tensor_shape) { in PromoteTypeToComplex()
499 Eigen::array<Eigen::DenseIndex, x_rank> tensor_shape = GetTensorShape<x_rank>(x_shape); in DCTCompute() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dnon_zero_ext.cc46 ShapeVector tensor_shape = {x_num}; in InferShape() local
88 ShapeVector tensor_shape = {x_num}; in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/proto/
Dtensor.proto26 TensorShape tensor_shape = 1; field
/third_party/mindspore/mindspore-src/source/mindspore/core/ir/
Dmeta_tensor_extends.cc28 auto tensor_shape = tens->shape(); in ToAbstract() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_frontend_func_impl/
Dtuple_to_tensor.cc53 std::vector<int64_t> tensor_shape = {0}; in CreateEmptyTupleTensorByType() local
64 std::vector<int64_t> tensor_shape = {SizeToLong(new_values.size())}; in CreateTensorByTupleCast() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tflite/
Dtflite_split_v_parser.cc52 auto tensor_shape = tensor->shape; in Parse() local
Dtflite_split_parser.cc44 const auto tensor_shape = shape_tensor->shape; in Parse() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/control/
Dtensorlist_fromtensor_infer.c52 vvector tensor_shape; in TensorListFromTensorInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Ddct_cpu_kernel.cc65 Eigen::array<Eigen::DenseIndex, x_rank> tensor_shape; in GetTensorShape() local
161 Eigen::array<Eigen::DenseIndex, x_rank> tensor_shape; in ConstructW4Tensor() local
187 Eigen::array<Eigen::DenseIndex, x_rank> tensor_shape; in ConstructW4TensorForComplex() local
262 Eigen::array<Eigen::DenseIndex, x_rank> tensor_shape; in OrthoNormalize() local
324 …nsor<T, x_rank, Eigen::RowMajor> in_tensor, Eigen::array<Eigen::DenseIndex, x_rank> tensor_shape) { in PromoteTypeToComplex()
435 Eigen::array<Eigen::DenseIndex, x_rank> tensor_shape = GetTensorShape<x_rank>(x_shape); in DCTCompute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dtensor_layout.cc50 const Arrangement &tensor_shape) { in Init()
73 const Shape &tensor_shape) { in InitFromVector()
112 … const Shape &tensor_shape, bool interleaved_parallel, bool check_device_num) { in InitFromExtendVector()
634 Shape tensor_shape(tensor_shape_origin_.array()); in TransferRepeatLayout() local
Dprime_generator.cc46 int64_t PrimeGenerator::GetCoprimeNum(const Shape &tensor_shape) { in GetCoprimeNum()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/
Dtensorlist.cc77 …ensorList::MallocTensorListData(TypeId dtype, const std::vector<std::vector<int> > &tensor_shape) { in MallocTensorListData()
302 std::vector<std::vector<int> > tensor_shape{}; in CopyTensorList() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/utils/
Dfft_helper.cc24 int64_t GetCalculateElementNum(std::vector<int64_t> tensor_shape, std::vector<int64_t> dim, std::ve… in GetCalculateElementNum()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/
Dtensorlist_c_utils.c19 int MallocTensorListData(TensorListC *tensor_list, TypeIdC dtype, const vvector *tensor_shape) { in MallocTensorListData()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/utils/
Dfft_helper.cc37 int64_t GetCalculateElementNum(std::vector<int64_t> tensor_shape, std::vector<int64_t> dim, std::ve… in GetCalculateElementNum()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_tensor.h43 Shape4D tensor_shape; member
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/control/
Dtensorlist_fromtensor.cc36 std::vector<int> tensor_shape = input0_->shape(); in IsCompatibleShape() local

12345