Home
last modified time | relevance | path

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

12

/third_party/mindspore/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
Dredistribution_operator_infer_test.cc75 Shape tensor_shape = dev_mat; in InferOperatorCheckAll() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/proto/
Dtensor.proto26 TensorShape tensor_shape = 1; field
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dconstruct_operator.h46 void UpdateTensorShape(const Shape &tensor_shape) { tensor_shape_ = tensor_shape; } in UpdateTensorShape()
Dtensor_layout.cc49 const Arrangement &tensor_shape) { in Init()
72 const Shape &tensor_shape) { in InitFromVector()
411 Shape tensor_shape(tensor_shape_origin_.array()); in TransferRepeatLayout() local
/third_party/mindspore/mindspore/core/ir/
Dmeta_tensor_extends.cc35 auto tensor_shape = tens->shape(); in ToAbstract() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_split_v_parser.cc55 auto tensor_shape = tensor->shape; in Parse() local
Dtflite_split_parser.cc47 const auto tensor_shape = shape_tensor->shape; in Parse() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/control/
Dtensorlist_fromtensor_infer.c50 vvector tensor_shape; in TensorListFromTensorInferShape() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_tensor.h35 Shape4D tensor_shape; member
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/control/
Dtensorlist_fromtensor.cc36 std::vector<int> tensor_shape = input0_->shape(); in IsCompatibleShape() local
Dtensor_array.cc61 std::vector<std::vector<int>> tensor_shape(shape.front(), element_shape_v); in Init() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dtf_model_parser.cc98 auto &tensor_shape = tensor_proto.tensor_shape(); in GetShapeSize() local
109 auto &tensor_shape = tensor_proto.tensor_shape(); in SetFloatTensorInfo() local
144 auto &tensor_shape = tensor_proto.tensor_shape(); in SetInt32TensorInfo() local
179 auto &tensor_shape = tensor_proto.tensor_shape(); in SetInt64TensorInfo() local
222 auto &tensor_shape = tensor_proto.tensor_shape(); in SetBoolTensorInfo() local
249 auto &tensor_shape = tensor_proto.tensor_shape(); in SetStringTensorInfo() local
399 const tensorflow::TensorShapeProto &tensor_shape = tensor_proto.tensor_shape(); in ConvertConstTensor() local
/third_party/mindspore/third_party/proto/tensorflow/
Dtensor.proto20 TensorShapeProto tensor_shape = 2; field
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/
Dfill.cc61 auto tensor_shape = in_tensors_[0]->shape(); in RunShape() local
/third_party/mindspore/mindspore/lite/src/delegate/nnrt/
Dnnrt_model_kernel.cc102 auto tensor_shape = tensor.Shape(); in PrepareInputs() local
/third_party/mindspore/mindspore/lite/src/runtime/
Dinfer_manager.cc167 auto tensor_shape = std::vector<std::vector<int>>( in KernelInferShape() local
/third_party/mindspore/mindspore/lite/tools/converter/legacy_optimizer/graph/
Dinfershape_pass.cc83 std::vector<int32_t> tensor_shape{}; in ConvertTensorList() local
138 auto tensor_shape = tensorT->dims; in ConvertString() local
166 auto tensor_shape = tensorT->dims; in ConvertOtherTensor() local
/third_party/mindspore/mindspore/lite/src/cxx_api/tensor/
Dtensor_impl.h132 std::vector<int> tensor_shape; in SetShape() local
/third_party/mindspore/mindspore/lite/src/
Dtensorlist.cc83 …ensorList::MallocTensorListData(TypeId dtype, const std::vector<std::vector<int> > &tensor_shape) { in MallocTensorListData()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/data/
Ddata_utils.cc688 std::vector<int64_t> tensor_shape = input->shape().AsVector(); in BatchTensorToCVTensorVector() local
714 std::vector<int64_t> tensor_shape = input->shape().AsVector(); in BatchTensorToTensorVector() local
738 std::vector<int64_t> tensor_shape = input.front()->shape().AsVector(); in TensorVectorToBatchTensor() local

12