Searched refs:kPrimTensorListFromTensor (Results 1 – 5 of 5) sorted by relevance
154 …_from_tensor = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTensorListFromTensor)); in DefineFowardPattern()224 …_from_tensor = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTensorListFromTensor)); in DefinebackwardPattern()
426 if (!utils::isa<CNodePtr>(input) || !CheckPrimitiveType(input, prim::kPrimTensorListFromTensor)) { in CreateLSTMNode()
593 if (!utils::isa<CNodePtr>(input) || !CheckPrimitiveType(input, prim::kPrimTensorListFromTensor)) { in CreateLSTMNode()
45 return opt::CheckPrimitiveType(anf_node, prim::kPrimTensorListFromTensor) || in IsTensorListOp()
457 inline const PrimitivePtr kPrimTensorListFromTensor = std::make_shared<Primitive>("TensorListFromTe… variable