Home
last modified time | relevance | path

Searched defs:tensor_x (Results 1 – 25 of 38) sorted by relevance

12

/third_party/mindspore/tests/ut/python/pynative_mode/
Dtest_outermost_non_tensor_input.py31 def construct(self, tuple_a, tensor_x, list_b, tensor_y, scalar, dict_c, flag): argument
44 def construct(self, tuple_a, tensor_x, list_b, tensor_y, scalar, dict_c, flag): argument
55 def construct(self, tuple_a, tensor_x, list_b, tensor_y, tensor_z, dict_c): argument
87 def construct(self, tensor_x, tuple_a, list_b, tensor_y, tensor_z, dict_c): argument
/third_party/mindspore/tests/ut/python/pipeline/parse/
Dtest_outermost_net_pass_non_tensor_inputs.py31 def construct(self, tuple_a, tensor_x, list_b, tensor_y, scalar, dict_c, flag): argument
44 def construct(self, tuple_a, tensor_x, list_b, tensor_y, scalar, dict_c, flag): argument
55 def construct(self, tuple_a, tensor_x, list_b, tensor_y, tensor_z, dict_c): argument
87 def construct(self, tensor_x, tuple_a, list_b, tensor_y, tensor_z, dict_c): argument
Dtest_ms_function_pass_non_tensor_inputs.py49 def tensor_reduce(tensor_x, axis, tensor_y): argument
/third_party/mindspore/tests/ut/cpp/ops/
Dtest_ops_argmax.cc37 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{2,… in TEST_F() local
65 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat16, std::vector<int64_t>{2,… in TEST_F() local
Dtest_ops_prelu.cc36 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{2,… in TEST_F() local
65 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat16, std::vector<int64_t>{5,… in TEST_F() local
Dtest_ops_reshape.cc36 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{2,… in TEST_F() local
64 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat16, std::vector<int64_t>{2,… in TEST_F() local
Dtest_ops_l2normalize.cc37 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{2,… in TEST_F() local
Dtest_ops_rfft.cc37 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{1,… in TEST_F() local
Dtest_ops_fftimag.cc35 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{1,… in TEST_F() local
Dtest_ops_batchtospace.cc40 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{12… in TEST_F() local
Dtest_ops_sin.cc35 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{1,… in TEST_F() local
Dtest_ops_flatten.cc35 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{1,… in TEST_F() local
Dtest_ops_neg.cc36 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{2,… in TEST_F() local
Dtest_ops_cos.cc37 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{2,… in TEST_F() local
Dtest_ops_depthtospace.cc37 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{3,… in TEST_F() local
Dtest_ops_elu.cc38 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{2,… in TEST_F() local
Dtest_ops_sqrt.cc35 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{1,… in TEST_F() local
Dtest_ops_fftreal.cc36 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{2,… in TEST_F() local
Dtest_ops_exp.cc36 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat16, std::vector<int64_t>{2,… in TEST_F() local
Dtest_ops_leakyrelu.cc36 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{3,… in TEST_F() local
Dtest_ops_log.cc35 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{1,… in TEST_F() local
Dtest_ops_floor.cc37 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{2,… in TEST_F() local
Dtest_ops_add.cc35 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{1,… in TEST_F() local
Dtest_ops_pow.cc35 …auto tensor_x = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{1,… in TEST_F() local
/third_party/mindspore/tests/st/pynative/non_tensor_input/
Dtest_pynative_outermost_non_tensor.py34 def construct(self, tensor_x, tuple_a, list_b, tensor_y, tensor_z, dict_c): argument
50 def construct(self, tuple_a, tensor_x, list_b, tensor_y, tensor_z, dict_c): argument

12