Searched refs:kPrimTensorListStack (Results 1 – 3 of 3) sorted by relevance
176 …tensor_list_stack = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTensorListStack)); in DefineFowardPattern()246 …tensor_list_stack = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTensorListStack)); in DefinebackwardPattern()
302 …tensor_list_stack = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTensorListStack)); in DefinePattern()
459 inline const PrimitivePtr kPrimTensorListStack = std::make_shared<Primitive>("TensorListStack"); variable