Home
last modified time | relevance | path

Searched refs:Squeeze (Results 1 – 25 of 117) sorted by relevance

12345

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dsqueeze_gpu_kernel.cc21 MS_REG_GPU_KERNEL_ONE(Squeeze, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberT…
23 MS_REG_GPU_KERNEL_ONE(Squeeze, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberT…
25 MS_REG_GPU_KERNEL_ONE(Squeeze, KernelAttr().AddInputAttr(kNumberTypeFloat64).AddOutputAttr(kNumberT…
27 MS_REG_GPU_KERNEL_ONE(Squeeze, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberType…
29 MS_REG_GPU_KERNEL_ONE(Squeeze, KernelAttr().AddInputAttr(kNumberTypeInt16).AddOutputAttr(kNumberTyp…
31 MS_REG_GPU_KERNEL_ONE(Squeeze, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTyp…
33 MS_REG_GPU_KERNEL_ONE(Squeeze, KernelAttr().AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumberTyp…
35 MS_REG_GPU_KERNEL_ONE(Squeeze, KernelAttr().AddInputAttr(kNumberTypeUInt8).AddOutputAttr(kNumberTyp…
37 MS_REG_GPU_KERNEL_ONE(Squeeze, KernelAttr().AddInputAttr(kNumberTypeUInt16).AddOutputAttr(kNumberTy…
39 MS_REG_GPU_KERNEL_ONE(Squeeze, KernelAttr().AddInputAttr(kNumberTypeUInt32).AddOutputAttr(kNumberTy…
[all …]
/third_party/mindspore/mindspore/core/ops/
Dsqueeze.h36 class MS_CORE_API Squeeze : public PrimitiveC {
39 Squeeze() : PrimitiveC(kNameSqueeze) { InitIOName({"x"}, {"output"}); } in Squeeze() function
41 ~Squeeze() = default;
42 MS_DECLARE_PARENT(Squeeze, PrimitiveC);
55 using PrimSqueezePtr = std::shared_ptr<Squeeze>;
Dsqueeze.cc21 void Squeeze::Init(const std::vector<int64_t> &axis) { set_axis(axis); } in Init()
22 void Squeeze::set_axis(const std::vector<int64_t> &axis) { (void)AddAttr(kAxis, MakeValue(axis)); } in set_axis()
23 std::vector<int64_t> Squeeze::get_axis() const { return GetValue<std::vector<int64_t>>(GetAttr(kAxi… in get_axis()
70 REGISTER_PRIMITIVE_C(kNameSqueeze, Squeeze);
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dreshape_cpu_kernel.h123 MS_REG_CPU_KERNEL(Squeeze, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberTypeInt8…
124 MS_REG_CPU_KERNEL(Squeeze, KernelAttr().AddInputAttr(kNumberTypeInt16).AddOutputAttr(kNumberTypeInt…
126 MS_REG_CPU_KERNEL(Squeeze, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeInt…
128 MS_REG_CPU_KERNEL(Squeeze, KernelAttr().AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumberTypeInt…
130 MS_REG_CPU_KERNEL(Squeeze, KernelAttr().AddInputAttr(kNumberTypeBool).AddOutputAttr(kNumberTypeBool…
131 MS_REG_CPU_KERNEL(Squeeze, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTypeF…
133 MS_REG_CPU_KERNEL(Squeeze, KernelAttr().AddInputAttr(kNumberTypeUInt8).AddOutputAttr(kNumberTypeUIn…
135 MS_REG_CPU_KERNEL(Squeeze, KernelAttr().AddInputAttr(kNumberTypeUInt16).AddOutputAttr(kNumberTypeUI…
137 MS_REG_CPU_KERNEL(Squeeze, KernelAttr().AddInputAttr(kNumberTypeUInt32).AddOutputAttr(kNumberTypeUI…
139 MS_REG_CPU_KERNEL(Squeeze, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUI…
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/
Darray_ops_declare.cc87 INPUT_MAP(Squeeze) = {{1, INPUT_DESC(x)}};
88 ATTR_MAP(Squeeze) = {{"axis", ATTR_DESC(axis, AnyTraits<int64_t>(), AnyTraits<std::vector<int64_t>>…
89 OUTPUT_MAP(Squeeze) = {{0, OUTPUT_DESC(y)}};
90 REG_ADPT_DESC(Squeeze, prim::kPrimSqueeze->name(), ADPT_DESC(Squeeze))
Darray_ops_declare.h45 DECLARE_OP_ADAPTER(Squeeze)
46 DECLARE_OP_USE_OUTPUT(Squeeze)
/third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/src/
Dyolo.py265 intersect_area = P.Squeeze(-1)(intersect_wh[:, :, :, :, :, 0:1]) * \
266 P.Squeeze(-1)(intersect_wh[:, :, :, :, :, 1:2])
267 …box1_area = P.Squeeze(-1)(box1_wh[:, :, :, :, :, 0:1]) * P.Squeeze(-1)(box1_wh[:, :, :, :, :, 1:2])
268 …box2_area = P.Squeeze(-1)(box2_wh[:, :, :, :, :, 0:1]) * P.Squeeze(-1)(box2_wh[:, :, :, :, :, 1:2])
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dsqueeze_int8.cc100 auto Squeeze = reinterpret_cast<SqueezeInt8CPUKernel *>(cdata); in SqueezeInt8Run() local
101 Squeeze->DoExecute(task_id); in SqueezeInt8Run()
/third_party/mindspore/tests/st/model_zoo_tests/yolov3/src/
Dyolov3.py198 self.squeeze = P.Squeeze(axis=(2, 3))
477 intersect_area = P.Squeeze(-1)(intersect_wh[:, :, :, :, :, 0:1]) * \
478 P.Squeeze(-1)(intersect_wh[:, :, :, :, :, 1:2])
479 …box1_area = P.Squeeze(-1)(box1_wh[:, :, :, :, :, 0:1]) * P.Squeeze(-1)(box1_wh[:, :, :, :, :, 1:2])
480 …box2_area = P.Squeeze(-1)(box2_wh[:, :, :, :, :, 0:1]) * P.Squeeze(-1)(box2_wh[:, :, :, :, :, 1:2])
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_squeeze_parser.cc26 auto prim = std::make_unique<ops::Squeeze>(); in Parse()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/core/
Dcv_tensor.cc129 void CVTensor::Squeeze() { in Squeeze() function in mindspore::dataset::CVTensor
130 Tensor::Squeeze(); in Squeeze()
Dcv_tensor.h93 void Squeeze() override;
/third_party/mindspore/mindspore/_extends/graph_kernel/expanders/
Dsqueeze.py20 class Squeeze(Expander): class
D__init__.py62 from .squeeze import Squeeze
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/data/
Done_hot_op.cc34 inputs_copy.push_back(inputs[0].Squeeze()); in OutputShape()
Dconcatenate_op.cc35 inputs_copy.push_back(inputs[0].Squeeze()); in OutputShape()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dtf_squeeze_parser.cc29 auto prim = std::make_unique<ops::Squeeze>(); in Parse()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_squeeze_parser.cc32 auto prim = std::make_unique<ops::Squeeze>(); in Parse()
/third_party/mindspore/mindspore/lite/src/delegate/npu/op/
Dsqueeze_npu.cc21 squeeze_ = new (std::nothrow) hiai::op::Squeeze(name_); in Init()
Dsqueeze_npu.h47 hiai::op::Squeeze *squeeze_ = nullptr;
/third_party/mindspore/tests/st/ops/graph_kernel/
Dtest_squeeze.py26 self.squeeze = P.Squeeze(axis)
/third_party/mindspore/mindspore/nn/layer/
Dpooling.py218 self.squeeze = P.Squeeze(2)
384 self.squeeze = P.Squeeze(2)
/third_party/mindspore/mindspore/nn/probability/distribution/
Dcategorical.py157 self.squeeze = P.Squeeze()
158 self.squeeze_first_axis = P.Squeeze(0)
159 self.squeeze_last_axis = P.Squeeze(-1)
/third_party/mindspore/mindspore/boost/
Dadasum.py59 recv_part = P.Squeeze()(recv_part)
91 delta_w = P.Squeeze()(delta_w)
130 recv_part = P.Squeeze()(recv_part)
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Dsqueeze_infer.c68 REG_INFER(Squeeze, PrimType_Squeeze, SqueezeInferShape)

12345