Searched defs:tensor_rank (Results 1 – 4 of 4) sorted by relevance
390 def tensor_rank(self, name="tensor_rank"): member in LinearOperator
492 uint32_t tensor_rank = static_cast<uint32_t>(tensor->dims->size); in ConvertTensorTypeToNNType() local1248 const uint32_t tensor_rank = static_cast<uint32_t>(dims->size); in AddNewInputConstantTensor() local1546 uint32_t tensor_rank = static_cast<uint32_t>(tensor->dims->size); in AddTensor() local
623 int32_t tensor_rank = result_type.getShape().size(); in convertConcatV2Op() local
4663 int64_t tensor_rank = tensor_ty.getRank(); in matchAndRewrite() local