Home
last modified time | relevance | path

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

12345

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/expanders/
Dmatmul_fusion.cc43 ShapeVector new_shape(a->shape.size() - b->shape.size(), 1); in Expand() local
47 ShapeVector new_shape(b->shape.size() - a->shape.size(), 1); in Expand() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/dataset/
Drandom_horizontal_flip_op_test.cc58 TensorShape new_shape({frame_num, shape[-3], shape[-2], shape[-1]}); in TEST_F() local
85 TensorShape new_shape({frame_num, shape[-3], shape[-2], shape[-1]}); in TEST_F() local
111 TensorShape new_shape({temp->Size()}); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/expanders/
Dreshape.cc31 ShapeVector new_shape = shape; in InferShape() local
77 ShapeVector new_shape; in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/
Drandom_data_op.cc71 std::unique_ptr<TensorShape> new_shape; in GenerateSchema() local
116 std::unique_ptr<TensorShape> new_shape = nullptr; in CreateRandomRow() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dmap.cc83 Shape new_shape; in ExpandMapByNone() local
110 Shape new_shape; in ExpandMapByDecreaseNumber() local
Darrangement.cc99 Shape new_shape; in GetExpandedShapeByExpandListRemoveLeft() local
184 Shape new_shape; in GetExpandedShapeByExpandListReserveLeft() local
/third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/
Dtest_sparse_reshape_op.py28 def construct(self, indices, shape, new_shape): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/image/
Dhorizontal_flip_op.cc34 TensorShape new_shape({num_batch, input_shape[-3], input_shape[-2], input_shape[-1]}); in Compute() local
Dvertical_flip_op.cc35 TensorShape new_shape({num_batch, input_shape[-3], input_shape[-2], input_shape[-1]}); in Compute() local
Drandom_horizontal_flip_op.cc45 TensorShape new_shape({num_batch, input_shape[-3], input_shape[-2], input_shape[-1]}); in Compute() local
Dlite_image_utils.cc522 TensorShape new_shape = TensorShape({output_height, output_width, 3}); in Resize() local
630 TensorShape new_shape = TensorShape({height, width, input->shape()[2]}); in ResizePreserve() local
676 TensorShape new_shape = TensorShape({input_height, input_width, kInputChannel}); in RgbToBgr() local
712 TensorShape new_shape = TensorShape({input_height, input_width, 1}); in RgbToGray() local
769 TensorShape new_shape = TensorShape({pad_height, pad_width, input->shape()[2]}); in Pad() local
840 TensorShape new_shape = TensorShape({height, width, input->shape()[2]}); in RotateAngleWithOutMirror() local
915 TensorShape new_shape = TensorShape({height, width, input->shape()[2]}); in RotateAngleWithMirror() local
1035 TensorShape new_shape = TensorShape({height, width, input->shape()[2]}); in Affine() local
1200 TensorShape new_shape = TensorShape({input_channel, input_height, input_width}); in HwcToChw() local
Dnormalize_op.cc53 TensorShape new_shape({num_batch, input_shape[-3], input_shape[-2], input_shape[-1]}); in Compute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/sparse/
Dsparse_reshape_gpu_kernel.cc84 int64_t *new_shape = GetDeviceAddress<int64_t>(inputs, kIndex2); in LaunchKernel() local
130 int SparseReshapeGpuKernelMod::CalShape(const int64_t *shape, const int64_t *new_shape, int64_t *y_… in CalShape()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/
Dflatten_ext.cc38 auto new_shape = std::make_shared<ValueTuple>(out_shape); in FlattenExtAscendCustomize() local
/third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/
Dtest_sparse_reshape_op.py29 def construct(self, indices, shape, new_shape): argument
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/view/
Dunsqueeze_strides_calc.cc36 auto new_shape = old_shape; in UnsqueezeCalc() local
Dselectview_strides_calc.cc46 auto new_shape = old_shape; in SelectViewCalc() local
Dexpand_dims_strides_calc.cc36 auto new_shape = old_shape; in ExpandDimsCalc() local
Dslice_ext_strides_calc.cc73 auto new_shape = old_shape; in SliceExtCalc() local
Dslice_strides_calc.cc80 auto new_shape = size; in SliceCalc() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/pass/format_pass/
Dpass_utils.cc58 …auto new_shape = TransShape(src_tensor->shape(), {src_tensor->format(), dst_tensor->format()}, &re… in SetShape() local
84 auto new_shape = TransShape(tensor->shape(), {tensor->format(), dst_format}, &ret); in TransTensorShapeAndFormat() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Ddropout_do_mask_info.cc197 ValuePtr new_shape = MakeValue(input_slice_shape); in SetGenMaskShape() local
243 ValuePtr new_shape = MakeValue(input_slice_shape); in GetDropoutGenMaskReplaceOp() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/api/
Dtensor_c_test.cc51 int64_t new_shape[new_shape_num] = {1, 2, 3, 1}; in TEST_F() local
/third_party/mindspore/mindspore-src/source/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-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/
Dshuffle_tensorrt.cc166 std::vector<int64_t> new_shape(squeeze_shape.d, squeeze_shape.d + squeeze_shape.nbDims); in AddSqueezeOp() local
346 …auto new_shape = ctx_->network()->addElementWise(*x, *y, nvinfer1::ElementWiseOperation::kSUM)->ge… in AddBroadcastToOp() local

12345