Home
last modified time | relevance | path

Searched defs:data_shape (Results 1 – 25 of 60) sorted by relevance

123

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dget_squeeze_slice_shape.cc33 inline size_t GetOutDim(const PrimitivePtr &primitive, const ShapeVector &data_shape) { in GetOutDim()
52 auto data_shape = shape_map[kShape]; in GetSqueezeSliceShapeInferInner() local
69 auto data_shape = data_abs->GetShape()->GetShapeVector(); in InferShape() local
77 auto data_shape = data_abs->GetShape()->GetShapeVector(); in InferType() local
Dnormalize_tuple_index.cc39 size_t NormalizeTupleIndex::NormalizeDimIndex(const ShapeVector &data_shape, size_t dim_index, in NormalizeDimIndex()
75 AbstractBasePtr NormalizeIntIndex(const ShapeVector &data_shape, const AbstractBasePtr &index_val_a… in NormalizeIntIndex()
92 AbstractBasePtr NormalizeSequenceIndex(const ShapeVector &data_shape, const AbstractBasePtr &index_… in NormalizeSequenceIndex()
124 AbstractBasePtr NormalizeNoneIndex(const ShapeVector &data_shape, size_t dim_index, in NormalizeNoneIndex()
144 AbstractBasePtr NormalizeBoolSequenceIndex(const ShapeVector &data_shape, const AbstractBasePtr &in… in NormalizeBoolSequenceIndex()
181 AbstractBasePtr NormalizeEllipsisIndex(const ShapeVector &data_shape, size_t dim_index, in NormalizeEllipsisIndex()
222 auto data_shape = shape_map[kShape]; in NormalizeTupleIndexInferInner() local
256 auto data_shape = shape_map[kShape]; in InferShape() local
Dellipsis_to_slice.cc35 auto data_shape = input_args[0]->GetShape()->cast<abstract::ShapePtr>()->shape(); in EllipsisToSliceInferShape() local
56 auto data_shape = input_args[0]->GetShape()->cast<abstract::ShapePtr>()->shape(); in InferType() local
70 auto data_shape = input_args[0]->GetShape()->cast<abstract::ShapePtr>()->shape(); in InferShapeAndType() local
Dslice_to_indices.cc58 …asePtr ConstSliceToIndices(const std::vector<int64_t> &init_by_none, const ShapeVector &data_shape, in ConstSliceToIndices()
108 std::vector<int64_t> CalSliceToIndices(const ShapeVector &data_shape, size_t index_axis, int64_t ex… in CalSliceToIndices()
155 ShapeVector data_shape = input_args[0]->GetShape()->GetShapeVector(); in SliceToIndicesInferInner() local
183 auto data_shape = input_args[kInputIndex0]->GetShape()->GetShapeVector(); in InferShape() local
Dnormalize_slice.cc40 …BasePtr NormalizeSliceInfo(const std::vector<int64_t> &init_by_none, const ShapeVector &data_shape, in NormalizeSliceInfo()
81 …ShapeVector data_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->GetShape()… in NormalizeSliceInferInner() local
Dremake_tuple_index.cc44 …ShapeVector data_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(data_abs->GetShape())[kSh… in RemakeTupleIndexInferInner() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dnormalize_tuple_index_cpu_kernel.cc53 void NormalizeTupleIndexCpuKernelMod::NormalizeIntIndex(const ShapeVector &data_shape, int64_t *out… in NormalizeIntIndex()
64 void NormalizeTupleIndexCpuKernelMod::NormalizeSequenceIndex(const ShapeVector &data_shape, int64_t… in NormalizeSequenceIndex()
83 void NormalizeTupleIndexCpuKernelMod::NormalizeBoolSequenceIndex(const ShapeVector &data_shape, int… in NormalizeBoolSequenceIndex()
113 …lizeTupleIndexCpuKernelMod::NormalizeNoneIndex(int64_t *output_addr, const ShapeVector &data_shape, in NormalizeNoneIndex()
126 …TupleIndexCpuKernelMod::NormalizeEllipsisIndex(int64_t *output_addr, const ShapeVector &data_shape, in NormalizeEllipsisIndex()
162 const ShapeVector &data_shape = data_shapes_[0]; in LaunchKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/
Dtensor_index_py.cc291 std::vector<TensorIndex> TensorIndex::TransformEllipsisToSlice(const ShapeVector &data_shape, in TransformEllipsisToSlice()
460 static bool CheckLargeTensor(const ShapeVector &data_shape) { in CheckLargeTensor()
467 py::tuple TensorIndex::GenerateNonZeroIndex(const ShapeVector &data_shape, const TensorPtr &tensor_… in GenerateNonZeroIndex()
489 std::vector<TensorPtr> TensorIndex::GenerateNonZeroIndexTensorList(const ShapeVector &data_shape, in GenerateNonZeroIndexTensorList()
505 bool TensorIndex::TensorGetitemByTupleParseTensorIndex(const ShapeVector &data_shape, const TensorP… in TensorGetitemByTupleParseTensorIndex()
535 const ShapeVector &data_shape, const std::vector<TensorIndex> &tuple_index) { in GetStrideInfoFromTuple()
607 const ShapeVector &data_shape, const std::vector<TensorIndex> &index) { in GetExpandDimsInfo()
661 py::object TensorIndex::TensorGetitemByTuple(const ShapeVector &data_shape, const std::vector<Tenso… in TensorGetitemByTuple()
802 py::object TensorIndex::GenerateIndicesFromTupleOfTensor(const ShapeVector &data_shape, in GenerateIndicesFromTupleOfTensor()
863 TensorIndex TensorIndex::FormatIndex(const TensorIndex &idx, const ShapeVector &data_shape, size_t … in FormatIndex()
[all …]
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/c_api/
Dtensor_test.cc40 int64_t data_shape[] = {1, 1, 3, 3}; in TEST_F() local
83 int64_t data_shape[] = {1, 1, 2, 2}; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/symbol_ops_impl/
Dnormalize_slice.cc32 auto data_shape = input_as<ListSymbol>(kIndex0); in Eval() local
93 auto data_shape = b->GetInputShape(kIndex0); in NormalizeSliceValueBuilder() local
Dstrided_slice.cc65 auto data_shape = input_as<ListSymbol>(kIndex0); in Eval() local
202 auto data_shape = b->GetInputShape(kIndex0); in StridedSliceShapeBuilder() local
/third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/
Dtest_soft_shrink_op.py59 def test_soft_shrink(dtype, data_shape, lambd): argument
149 def test_softshrink_grad(data_shape): argument
Dtest_hshrink_op.py47 def test_hshrink(dtype, data_shape, lambd): argument
Dtest_hshrink_grad_op.py59 def test_hshrink_grad(dtype, data_shape, lambd): argument
Dtest_selu_op.py70 def test_selu(data_shape, data_type): argument
Dtest_mish_op.py64 def test_mish(data_shape, data_type): argument
/third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/
Dtest_matrix_determinant_op.py82 def test_matrix_determinant(data_shape, data_type): argument
107 def test_log_matrix_determinant(data_shape, data_type): argument
Dtest_hshrink_op.py46 def test_hshrink(dtype, data_shape, lambd): argument
Dtest_hshrink_grad_op.py46 def test_hshrink_grad(dtype, data_shape, lambd): argument
Dtest_non_zero_op.py42 def test_net(data_shape, data_type): argument
Dtest_selu_op.py69 def test_selu_op(data_type, data_shape): argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
Drl_ops.py125 def infer_shape(self, data_shape, count_shape, head_shape): argument
280 def infer_shape(self, data_shape, count_shape, head_shape, index_shape): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Dbitwise_ops_proto.cc31 Shape data_shape = op.GetInputDescByName("x").GetShape(); in IMPLEMT_INFERFUNC() local
/third_party/mindspore/mindspore-src/source/tests/st/ops/custom_ops_tbe/
Dcus_square.py33 def infer_shape(self, data_shape): argument
/third_party/mindspore/mindspore-src/source/tests/st/ops/dynamic_shape/
Dtest_dynamic_shape_prelu.py59 def test_dynamic_shape_prelu(data_shape, data_type): argument

123