Searched defs:tensor_base (Results 1 – 5 of 5) sorted by relevance
113 ShapePtr CheckShapeSame(const std::string &op, const AbstractTensorPtr &tensor_base, const Abstract… in CheckShapeSame()127 TypePtr CheckDtypeSame(const std::string &op, const AbstractTensorPtr &tensor_base, const AbstractT… in CheckDtypeSame()
103 AbstractTensorPtr tensor_base = CheckArg<AbstractTensor>(op_name, arg->elements(), 0); in InferImplStack() local935 AbstractTensorPtr tensor_base = nullptr; in InferImplConcat() local
205 for (const auto &tensor_base : tensor_base_list) { in SendTensorBase() local
155 TensorBase tensor_base = 1; field
941 TensorBase *tensor_base = tensor_summary_item.mutable_tensor_base(); in AddTensorStatInfo() local