Searched refs:tensor_w (Results 1 – 3 of 3) sorted by relevance
37 …auto tensor_w = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{3}… in TEST_F() local39 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() local68 MS_EXCEPTION_IF_NULL(tensor_w); in TEST_F()69 auto abstract = prelu->Infer({tensor_x->ToAbstract(), tensor_w->ToAbstract()}); in TEST_F()
57 …auto tensor_w = TensorConstructUtils::CreateOnesTensor(kNumberTypeFloat32, std::vector<int64_t>{64… in TEST_F() local59 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()
560 …int64_t tensor_w = static_cast<int64_t>(op.arguments[0].tensor_shape.shape_w * op.arguments[0].ten… in GetOptimalStr() local586 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() local666 if (tensor_w < 2 || tensor_w % 2 != 0) { in GetOptimalStr()