Home
last modified time | relevance | path

Searched refs:Flatten (Results 1 – 25 of 254) sorted by relevance

1234567891011

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dflatten_gpu_kernel.cc22 MS_REG_GPU_KERNEL_ONE(Flatten, KernelAttr().AddInputAttr(kNumberTypeFloat64).AddOutputAttr(kNumberT…
29 MS_REG_GPU_KERNEL_ONE(Flatten, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberT…
36 MS_REG_GPU_KERNEL_ONE(Flatten, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberT…
43 MS_REG_GPU_KERNEL_ONE(Flatten, KernelAttr().AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumberTyp…
50 MS_REG_GPU_KERNEL_ONE(Flatten, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTyp…
57 MS_REG_GPU_KERNEL_ONE(Flatten, KernelAttr().AddInputAttr(kNumberTypeInt16).AddOutputAttr(kNumberTyp…
65 MS_REG_GPU_KERNEL_ONE(Flatten, KernelAttr().AddInputAttr(kNumberTypeUInt32).AddOutputAttr(kNumberTy…
73 MS_REG_GPU_KERNEL_ONE(Flatten, KernelAttr().AddInputAttr(kNumberTypeUInt8).AddOutputAttr(kNumberTyp…
80 MS_REG_GPU_KERNEL_ONE(Flatten, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberType…
87 MS_REG_GPU_KERNEL_ONE(Flatten, KernelAttr().AddInputAttr(kNumberTypeBool).AddOutputAttr(kNumberType…
/third_party/mindspore/mindspore/core/ops/
Dflatten.h31 class MS_CORE_API Flatten : public PrimitiveC {
34 Flatten() : PrimitiveC(kNameFlatten) {} in Flatten() function
36 ~Flatten() = default;
37 MS_DECLARE_PARENT(Flatten, PrimitiveC);
43 using PrimFlattenPtr = std::shared_ptr<Flatten>;
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dreshape_cpu_kernel.h62 MS_REG_CPU_KERNEL(Flatten, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberTypeInt8…
63 MS_REG_CPU_KERNEL(Flatten, KernelAttr().AddInputAttr(kNumberTypeInt16).AddOutputAttr(kNumberTypeInt…
65 MS_REG_CPU_KERNEL(Flatten, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeInt…
67 MS_REG_CPU_KERNEL(Flatten, KernelAttr().AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumberTypeInt…
69 MS_REG_CPU_KERNEL(Flatten, KernelAttr().AddInputAttr(kNumberTypeBool).AddOutputAttr(kNumberTypeBool…
70 MS_REG_CPU_KERNEL(Flatten, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTypeF…
72 MS_REG_CPU_KERNEL(Flatten, KernelAttr().AddInputAttr(kNumberTypeUInt8).AddOutputAttr(kNumberTypeUIn…
74 MS_REG_CPU_KERNEL(Flatten, KernelAttr().AddInputAttr(kNumberTypeUInt16).AddOutputAttr(kNumberTypeUI…
76 MS_REG_CPU_KERNEL(Flatten, KernelAttr().AddInputAttr(kNumberTypeUInt32).AddOutputAttr(kNumberTypeUI…
78 MS_REG_CPU_KERNEL(Flatten, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUI…
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/
Dtransformation_ops_declare.cc22 INPUT_MAP(Flatten) = {{1, INPUT_DESC(x)}};
23 ATTR_MAP(Flatten) = EMPTY_ATTR_MAP;
24 OUTPUT_MAP(Flatten) = {{0, OUTPUT_DESC(y)}};
25 REG_ADPT_DESC(Flatten, prim::kPrimFlatten->name(), ADPT_DESC(Flatten))
Dtransformation_ops_declare.h35 DECLARE_OP_ADAPTER(Flatten)
36 DECLARE_OP_USE_OUTPUT(Flatten)
/third_party/mindspore/tests/st/ops/gpu/
Dtest_flatten_op.py28 self.flatten = P.Flatten()
37 self.flatten = P.Flatten()
50 self.flatten = P.Flatten()
65 self.flatten = P.Flatten()
/third_party/typescript/tests/baselines/reference/
DrecursiveConditionalTypes.js28 type Flatten<T extends readonly unknown[]> = T extends unknown[] ? _Flatten<T>[] : readonly _Flatte…
33 type B0 = Flatten<string[][][]>;
34 type B1 = Flatten<string[][] | readonly (number[] | boolean[][])[]>;
35 type B2 = Flatten<InfiniteArray<string>>;
183 declare type Flatten<T extends readonly unknown[]> = T extends unknown[] ? _Flatten<T>[] : readonly…
186 declare type B0 = Flatten<string[][][]>;
187 declare type B1 = Flatten<string[][] | readonly (number[] | boolean[][])[]>;
188 declare type B2 = Flatten<InfiniteArray<string>>;
DrecursiveConditionalTypes.symbols97 type Flatten<T extends readonly unknown[]> = T extends unknown[] ? _Flatten<T>[] : readonly _Flatte…
98 >Flatten : Symbol(Flatten, Decl(recursiveConditionalTypes.ts, 22, 1))
121 type B0 = Flatten<string[][][]>;
123 >Flatten : Symbol(Flatten, Decl(recursiveConditionalTypes.ts, 22, 1))
125 type B1 = Flatten<string[][] | readonly (number[] | boolean[][])[]>;
127 >Flatten : Symbol(Flatten, Decl(recursiveConditionalTypes.ts, 22, 1))
129 type B2 = Flatten<InfiniteArray<string>>;
131 >Flatten : Symbol(Flatten, Decl(recursiveConditionalTypes.ts, 22, 1))
DrecursiveConditionalTypes.types66 type Flatten<T extends readonly unknown[]> = T extends unknown[] ? _Flatten<T>[] : readonly _Flatte…
67 >Flatten : Flatten<T>
75 type B0 = Flatten<string[][][]>;
78 type B1 = Flatten<string[][] | readonly (number[] | boolean[][])[]>;
81 type B2 = Flatten<InfiniteArray<string>>;
/third_party/typescript/tests/cases/compiler/
DrecursiveConditionalTypes.ts31 type Flatten<T extends readonly unknown[]> = T extends unknown[] ? _Flatten<T>[] : readonly _Flatte… alias
36 type B0 = Flatten<string[][][]>;
37 type B1 = Flatten<string[][] | readonly (number[] | boolean[][])[]>;
38 type B2 = Flatten<InfiniteArray<string>>;
/third_party/boost/libs/hana/include/boost/hana/
Dflatten.hpp33 using Flatten = BOOST_HANA_DISPATCH_IF(flatten_impl<M>, in operator ()() typedef
42 return Flatten::apply(static_cast<Xs&&>(xs)); in operator ()()
/third_party/boost/boost/hana/
Dflatten.hpp33 using Flatten = BOOST_HANA_DISPATCH_IF(flatten_impl<M>, in operator ()() typedef
42 return Flatten::apply(static_cast<Xs&&>(xs)); in operator ()()
/third_party/mindspore/tests/ut/python/pipeline/parse/
Dtest_celllist.py22 from mindspore.nn import Flatten
55 self.seq = SequentialCell([AvgPool2d(3, 1), ReLU(), Flatten()])
/third_party/flutter/skia/third_party/externals/libwebp/src/enc/
Dpicture_tools_enc.c37 static void Flatten(uint8_t* ptr, int v, int stride, int size) { in Flatten() function
135 Flatten(y_ptr + x, values[0], y_stride, SIZE); in WebPCleanupTransparentArea()
136 Flatten(u_ptr + (x >> 1), values[1], uv_stride, SIZE2); in WebPCleanupTransparentArea()
137 Flatten(v_ptr + (x >> 1), values[2], uv_stride, SIZE2); in WebPCleanupTransparentArea()
/third_party/skia/third_party/externals/libwebp/src/enc/
Dpicture_tools_enc.c37 static void Flatten(uint8_t* ptr, int v, int stride, int size) { in Flatten() function
148 Flatten(y_ptr + x, values[0], y_stride, SIZE); in WebPCleanupTransparentArea()
149 Flatten(u_ptr + (x >> 1), values[1], uv_stride, SIZE2); in WebPCleanupTransparentArea()
150 Flatten(v_ptr + (x >> 1), values[2], uv_stride, SIZE2); in WebPCleanupTransparentArea()
/third_party/mindspore/tests/ut/python/utils/
Dtest_initializer_fuzz.py40 self.flatten = nn.Flatten()
82 self.flatten = nn.Flatten()
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_flatten_parser.cc25 auto prim = std::make_unique<ops::Flatten>(); in Parse()
/third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/
Dcaffe_flatten_parser.cc25 auto prim = std::make_unique<ops::Flatten>(); in Parse()
/third_party/mindspore/tests/ut/python/pynative_mode/nn/
Dtest_flatten.py25 flatten = nn.Flatten()
/third_party/skia/src/core/
DSkPaintPriv.h46 static void Flatten(const SkPaint& paint, SkWriteBuffer& buffer);
/third_party/mindspore/tests/ut/python/nn/
Dtest_flatten.py28 self.flatten = nn.Flatten()
/third_party/mindspore/tests/ut/python/exec/
Dtest_flatten.py28 self.flatten = P.Flatten()
/third_party/flutter/skia/src/core/
DSkPaintPriv.h53 static void Flatten(const SkPaint& paint, SkWriteBuffer& buffer);
DSkPicturePriv.h31 static void Flatten(const sk_sp<const SkPicture> , SkWriteBuffer& buffer);
/third_party/mindspore/tests/ut/python/train/summary/
Dtest_image_summary.py87 self.flatten = nn.Flatten()
108 self.flatten = nn.Flatten()

1234567891011