Home
last modified time | relevance | path

Searched refs:tensor_w (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/tests/ut/cpp/ops/
Dtest_ops_prelu.cc37 …auto tensor_w = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{3}… in TEST_F() local
39 MS_EXCEPTION_IF_NULL(tensor_w); in TEST_F()
40 auto abstract = prelu->Infer({tensor_x->ToAbstract(), tensor_w->ToAbstract()}); in TEST_F()
66 …auto tensor_w = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat16, std::vector<int64_t>{1}… in TEST_F() local
68 MS_EXCEPTION_IF_NULL(tensor_w); in TEST_F()
69 auto abstract = prelu->Infer({tensor_x->ToAbstract(), tensor_w->ToAbstract()}); in TEST_F()
Dtest_ops_conv2d.cc57 …auto tensor_w = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{64… in TEST_F() local
59 MS_EXCEPTION_IF_NULL(tensor_w); in TEST_F()
60 auto conv_abstract = conv_2d->Infer({tensor_x->ToAbstract(), tensor_w->ToAbstract()}); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_cost.cc560 …int64_t tensor_w = static_cast<int64_t>(op.arguments[0].tensor_shape.shape_w * op.arguments[0].ten… in GetOptimalStr() local
586 if (tensor_w < 2 || tensor_w % 2 != 0) { in GetOptimalStr()
644 …int64_t tensor_w = static_cast<int64_t>(op.arguments[0].tensor_shape.shape_w * op.arguments[0].ten… in GetOptimalStr() local
666 if (tensor_w < 2 || tensor_w % 2 != 0) { in GetOptimalStr()