Searched defs:npu_tensor (Results 1 – 1 of 1) sorted by relevance
51 std::vector<int64_t> GetNpuTensorShape(int dim, std::shared_ptr<hiai::AiTensor> npu_tensor) { in GetNpuTensorShape()68 bool IsSameShapeTensor(mindspore::MSTensor tensor, const std::shared_ptr<hiai::AiTensor> &npu_tenso… in IsSameShapeTensor()