Home
last modified time | relevance | path

Searched defs: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
66 …auto tensor_w = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat16, std::vector<int64_t>{1}… in TEST_F() local
Dtest_ops_conv2d.cc57 …auto tensor_w = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{64… in TEST_F() local
/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
644 …int64_t tensor_w = static_cast<int64_t>(op.arguments[0].tensor_shape.shape_w * op.arguments[0].ten… in GetOptimalStr() local