Searched refs:row_splits_shape (Results 1 – 3 of 3) sorted by relevance
171 TfLiteIntArray* row_splits_shape = TfLiteIntArrayCreate(1); in Prepare() local172 row_splits_shape->data[0] = input_size + 1; in Prepare()176 context->ResizeTensor(context, row_splits, row_splits_shape)); in Prepare()
1133 row_splits_shape = tensor_shape.TensorShape(1135 return tensor_spec.TensorSpec(row_splits_shape, self._dtype)
1311 TensorShape row_splits_shape; in FastParseExample() local1312 row_splits_shape.AddDim(serialized.size() + 1); in FastParseExample()1314 row_splits_shape); in FastParseExample()