Home
last modified time | relevance | path

Searched defs:x_tensor (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/
Dpooling_grad_fp32_tests.cc141 lite::Tensor x_tensor(TypeId::kNumberTypeFloat32, dim_x); in TEST_F() local
214 lite::Tensor x_tensor(TypeId::kNumberTypeFloat32, dim_x); in TEST_F() local
281 lite::Tensor x_tensor(TypeId::kNumberTypeFloat32, dim_x); in TEST_F() local
345 lite::Tensor x_tensor(TypeId::kNumberTypeFloat32, dim_x); in TEST_F() local
472 lite::Tensor x_tensor(TypeId::kNumberTypeFloat32, dim_x); in TEST_F() local
548 lite::Tensor x_tensor(TypeId::kNumberTypeFloat32, dim_x); in TEST_F() local
625 lite::Tensor x_tensor(TypeId::kNumberTypeFloat32, dim_x); in TEST_F() local
Ddeconvolution_grad_fp32_tests.cc81 lite::Tensor x_tensor(TypeId::kNumberTypeFloat32, dim_x); in TEST_F() local
189 lite::Tensor x_tensor(TypeId::kNumberTypeFloat32, dim_x); in TEST_F() local
294 lite::Tensor x_tensor(TypeId::kNumberTypeFloat32, dim_x); in TEST_F() local
398 lite::Tensor x_tensor(TypeId::kNumberTypeFloat32, dim_x); in TEST_F() local
505 lite::Tensor x_tensor(TypeId::kNumberTypeFloat32, dim_x); in TEST_F() local
615 lite::Tensor x_tensor(TypeId::kNumberTypeFloat32, dim_x); in TEST_F() local
Dbn_grad_fp32_test.cc60 auto x_tensor = CreateInTensor("./bngrad/input_x_2_4_5_3.bin", {batch, height, width, channels}); in TEST_F() local
131 auto x_tensor = CreateInTensor("./bngrad/input_x_2_4_5_3.bin", {batch, height, width, channels}); in TEST_F() local
Dconvolution_grad_fp32_tests.cc102 lite::Tensor x_tensor(TypeId::kNumberTypeFloat32, dim_x); in TEST_F() local
256 lite::Tensor x_tensor(TypeId::kNumberTypeFloat32, dim_x); in TEST_F() local
406 lite::Tensor x_tensor(TypeId::kNumberTypeFloat32, dim_x); in TEST_F() local
539 lite::Tensor x_tensor(TypeId::kNumberTypeFloat32, dim_x); in TEST_F() local
658 lite::Tensor x_tensor(TypeId::kNumberTypeFloat32, dim_x); in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Dpower_infer.c27 const TensorC *x_tensor = inputs[0]; in PowerInferShape() local
/third_party/mindspore/tests/ut/cpp/optimizer/
Dlib_test.cc101 …tensor::TensorPtr x_tensor = std::make_shared<tensor::Tensor>(kFloat32->type_id(), std::vector<int… in TEST_F() local
184 tensor::TensorPtr x_tensor = std::make_shared<tensor::Tensor>(kFloat32->type_id(), shp); in TEST_F() local
214 tensor::TensorPtr x_tensor = std::make_shared<tensor::Tensor>(kFloat32->type_id(), shp); in TEST_F() local
224 tensor::TensorPtr x_tensor = std::make_shared<tensor::Tensor>(kFloat32->type_id(), shp); in TEST_F() local
280 tensor::TensorPtr x_tensor = std::make_shared<tensor::Tensor>(kFloat32->type_id(), shp); in TEST_F() local
302 tensor::TensorPtr x_tensor = std::make_shared<tensor::Tensor>(kFloat32->type_id(), shp); in TEST_F() local
323 tensor::TensorPtr x_tensor = std::make_shared<tensor::Tensor>(kFloat32->type_id(), shp); in TEST_F() local
518 tensor::TensorPtr x_tensor = std::make_shared<tensor::Tensor>(kFloat32->type_id(), shp); in TEST_F() local
/third_party/mindspore/mindspore/core/ops/
Dabs.cc71 auto x_tensor = x->cast<tensor::TensorPtr>(); in AbsInferValue() local
Dsquare.cc68 auto x_tensor = x->cast<tensor::TensorPtr>(); in SquareInferValue() local
Dselect.cc102 auto x_tensor = x->cast<tensor::TensorPtr>(); in SelectInferValue() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/
Ddeconvolution_grad_filter.cc40 auto *x_tensor = in_tensors_.at(1); in Init() local
Dconvolution_grad_filter.cc41 auto *x_tensor = in_tensors_.at(1); in ReSize() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/
Dconvolution_fp16_grad_filter.cc39 auto *x_tensor = in_tensors_.at(1); in ReSize() local
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Darithmetic_simplify.cc139 auto x_tensor = GetValueNode<tensor::TensorPtr>(x_->cast<ValueNodePtr>()); in operator ()() local
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/
Dprim.cc1550 auto x_tensor = dyn_cast<AbstractTensor>(x); in IsSubtypeArray() local