Searched refs:shape_rank (Results 1 – 4 of 4) sorted by relevance
39 const int shape_rank = SizeOfDimension(shape, 0); in ResizeOutputTensor() local40 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(shape_rank); in ResizeOutputTensor()43 for (int i = 0; i < shape_rank; i++) { in ResizeOutputTensor()
67 int64_t shape_rank = shape.size(); in createReshapeOp() local69 RankedTensorType::get({shape_rank}, rewriter.getIntegerType(64)); in createReshapeOp()
74 int64_t shape_rank = shape.size(); in createReshapeOp() local76 RankedTensorType::get({shape_rank}, rewriter->getIntegerType(64)); in createReshapeOp()
2256 int32 shape_rank = c->Value(shape_rank_dim); in ValidateSparseTensor() local2257 if (index_rank != shape_rank) { in ValidateSparseTensor()2259 ") and shape rank (", shape_rank, in ValidateSparseTensor()