Home
last modified time | relevance | path

Searched defs:new_shape (Results 1 – 25 of 34) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dmap.cc84 Shape new_shape; in ExpandMapByNone() local
111 Shape new_shape; in ExpandMapByDecreaseNumber() local
Darrangement.cc90 Shape new_shape; in GetExpandedShapeByExpandListRemoveLeft() local
175 Shape new_shape; in GetExpandedShapeByExpandListReserveLeft() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/
Drandom_data_op.cc78 std::unique_ptr<TensorShape> new_shape; in GenerateSchema() local
278 std::unique_ptr<TensorShape> new_shape = nullptr; in CreateRandomRow() local
Dmindrecord_op.cc277 auto new_shape = TensorShape(column.Shape()); in LoadTensorRow() local
287 auto new_shape = TensorShape(shapeDetails); in LoadTensorRow() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/
Dlite_image_utils.cc432 TensorShape new_shape = TensorShape({height, width, input->shape()[2]}); in ResizePreserve() local
476 TensorShape new_shape = TensorShape({output_height, output_width, 3}); in RgbToBgr() local
510 TensorShape new_shape = TensorShape({output_height, output_width, 1}); in RgbToGray() local
562 TensorShape new_shape = TensorShape({pad_height, pad_width, input->shape()[2]}); in Pad() local
627 TensorShape new_shape = TensorShape({height, width, input->shape()[2]}); in RotateAngleWithOutMirror() local
697 TensorShape new_shape = TensorShape({height, width, input->shape()[2]}); in RotateAngleWithMirror() local
762 TensorShape new_shape = TensorShape({height, width, input->shape()[2]}); in Affine() local
Dimage_utils.cc166 TensorShape new_shape = TensorShape({output_height, output_width, 3}); in Resize() local
689 TensorShape new_shape = TensorShape({target_height, target_width, 3}); in CropAndResize() local
877 TensorShape new_shape({input_cv->shape()[0], input_cv->shape()[1], input_cv->shape()[2] + 1}); in NormalizePad() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/expanders/
Dreshape.cc33 ShapeVector new_shape = shape; in InferShape() local
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dconv1d_weight_expanding_pass.cc38 std::vector<int64_t> new_shape(shape); in ExpandFilterShape() local
/third_party/mindspore/mindspore/lite/test/ut/src/api/
Dtensor_c_test.cc49 int64_t new_shape[new_shape_num] = {1, 2, 3, 1}; in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dreduce_eliminate.h77 auto new_shape = std::make_shared<ValueTuple>(elements); in operator() local
/third_party/mindspore/mindspore/lite/test/ohos_test/src/
Dtensor_test.cc59 int64_t new_shape[new_shape_num] = {1, 2, 3, 1}; variable
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Ddropout_do_mask_info.cc209 ValuePtr new_shape = MakeValue(input_slice_shape); in SetGenMaskShape() local
256 ValuePtr new_shape = MakeValue(input_slice_shape); in GetDropoutGenMaskReplaceOp() local
Duniform_real_info.cc136 ValuePtr new_shape = MakeValue(input_shape); in UpdateShape() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dpack_fission.cc52 std::vector<size_t> new_shape = output_shape; in CreateNewPack() local
Dgather_v2_ds_fission.cc62 ShapeVector new_shape(shape); in CreatePad() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/model/
Dop_node.cc313 auto new_shape = GetListInt(attrs.find("shape")->second); in InferShape() local
353 DShape new_shape = inputs[0]->shape; in InferShape() local
359 DShape new_shape; in InferShape() local
439 DShape new_shape; in InferShape() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/
Dconv1d_inout_adjust.cc72 std::vector<int64_t> new_shape(shape); in ExpandFilterShape() local
Dunify_format.cc233 std::vector<float> new_shape; in ConvertOnnxResizeForConstShape() local
/third_party/mindspore/mindspore/lite/src/delegate/tensorrt/op/
Dshuffle_tensorrt.cc197 std::vector<int64_t> new_shape(unsqueeze_shape.d, unsqueeze_shape.d + unsqueeze_shape.nbDims); in AddUnsqueezeOp() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/core/
Dtensor_shape.cc219 std::vector<dsize_t> new_shape(raw_shape_.size()); in Squeeze() local
/third_party/mindspore/mindspore/lite/src/delegate/tensorrt/
Dtensorrt_subgraph.cc455 std::vector<int64_t> new_shape = lite::ConvertMSShape(out_dims); in Execute() local
/third_party/boost/libs/multi_array/test/
Dgenerative_tests.hpp262 std::vector<size_type> new_shape(A.num_dimensions()); in modify() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/
Dbatch_op.cc191 TensorShape new_shape = first_shape.PrependDim(static_cast<int64_t>(batch_size)); in BatchRows() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Darithmetic_simplify.cc623 std::vector<int64_t> new_shape = {1}; in ReorganizeEmptyGraph() local
/third_party/mindspore/mindspore/numpy/
Darray_ops.py254 def reshape(x, new_shape): argument

12